re-enable test_reduce_0d_default int test case in test_dtype (#3336)

This commit is contained in:
chenyu 2024-02-07 05:30:14 -05:00 committed by GitHub
commit 02636ff62d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -322,7 +322,7 @@ class TestTypeSpec(unittest.TestCase):
def test_reduce_0d_default(self):
assert Tensor.ones([2,3,0]).sum(2).dtype == dtypes.default_float
# assert Tensor.ones([2,3,0], dtype=dtypes.int).sum(2).dtype == dtypes.int # requires reduceop acc fix
assert Tensor.ones([2,3,0], dtype=dtypes.int).sum(2).dtype == dtypes.int
@given(strat.sampled_from([dtypes.int8,dtypes.int16,dtypes.int32,dtypes.int64]), strat.sampled_from([dtypes.float16,dtypes.float32,dtypes.float64]))
def test_arange(self, default_int, default_float):