github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

update test_uop_resolve (#13777)

Browse Source 
plain @unittest.expectedFailure is too broad
This commit is contained in:
chenyu
2025-12-20 12:40:59 -05:00
committed by GitHub
parent 3db2104fb8
commit 733ef0452c
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/unit/test_uop_resolve.py
View File

@@ -95,21 +95,24 @@ class TestUOpResolve(unittest.TestCase):
x = UOp.variable("i", 1, 10)
self.assertFalse(x < x)
@unittest.expectedFailure
def test_x_lt_xp1(self):
x = UOp.variable("i", 1, 10)
self.assertTrue(x < (x+1))
u = x < (x+1)
# TODO: improve
with self.assertRaises(ValueError):
bool(u)
def test_and_true(self):
u = UOp.variable("b", False, True, dtypes.bool) & True
with self.assertRaises(ValueError):
u = UOp.variable("b", False, True, dtypes.bool) & True
self.assertFalse(u)
bool(u)
@unittest.expectedFailure
def test_var_cmp_range(self):
v = UOp.variable("i", 1, 10)
u = (v > 4) | (v < 6)
self.assertTrue(u)
# TODO: improve
with self.assertRaises(ValueError):
bool(u)
def test_var_cmp_assert(self):
with self.assertRaises(ValueError):