github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

clean up DTYPES_DICT [pr] (#7845)

Browse Source
This commit is contained in:
chenyu
2024-11-22 10:01:34 -05:00
committed by GitHub
parent 4453ab51e1
commit 40d7535eeb
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_dtype.py
View File

@@ -109,6 +109,9 @@ class TestDType(unittest.TestCase):
def test_dtypes_fields(self):
fields = dtypes.fields()
self.assertIn("float", fields)
self.assertIn("float32", fields)
self.assertEqual(len(fields), 24)
self.assertTrue(all(isinstance(value, DType) for value in fields.values()))
self.assertTrue(all(issubclass(_to_np_dtype(value), np.generic) for value in fields.values() if _to_np_dtype(value) is not None))