mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
14 lines
564 B
Python
14 lines
564 B
Python
import unittest
|
|
from tinygrad import dtypes, Device
|
|
from tinygrad.device import is_dtype_supported
|
|
|
|
@unittest.skipUnless(Device.DEFAULT=="NULL", "Don't run when testing non-NULL backends")
|
|
class TestNULLSupportsDTypes(unittest.TestCase):
|
|
def test_null_supports_ints_floats_bool(self):
|
|
dts = dtypes.ints + dtypes.floats + (dtypes.bool,)
|
|
not_supported = [dt for dt in dts if not is_dtype_supported(dt)]
|
|
self.assertFalse(not_supported, msg=f"expected these dtypes to be supported by NULL: {not_supported}")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|