Fix test

jackzhxng · jackzhxng · commit b10de682c630 · 2024-09-03T14:29:36.000-07:00
diff --git a/exir/tests/test_tensor.py b/exir/tests/test_tensor.py
@@ -171,7 +171,7 @@ def test_allocation_info_fails(self) -> None:
         )
         for test_case in test_cases:
             kwargs = test_case[0]
-            with self.assertRaisesRegex(ValueError, test_case[1], msg=f"{kwargs}"):
+            with self.assertRaisesRegex(Exception, test_case[1], msg=f"{kwargs}"):
                 make_allocation_info(**kwargs)
 
     def test_contiguous_stride_from_shape(self) -> None:

Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ def test_allocation_info_fails(self) -> None:`
`171`	`171`	`)`
`172`	`172`	`for test_case in test_cases:`
`173`	`173`	`kwargs = test_case[0]`
`174`		`- with self.assertRaisesRegex(ValueError, test_case[1], msg=f"{kwargs}"):`
	`174`	`+ with self.assertRaisesRegex(Exception, test_case[1], msg=f"{kwargs}"):`
`175`	`175`	`make_allocation_info(**kwargs)`
`176`	`176`
`177`	`177`	`def test_contiguous_stride_from_shape(self) -> None:`