You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
572 B
13 lines
572 B
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, "NULL")]
|
|
self.assertFalse(not_supported, msg=f"expected these dtypes to be supported by NULL: {not_supported}")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|
|
|