Files
tinygrad/test/testextra/test_mockgpu.py
nimlgen a647f3dd2c move mockgpu to tests [pr] (#8396)
* move mockgpu to tests

* linter

* i'm so sorry

* sorry, python

* path
2024-12-24 23:48:02 +03:00

13 lines
496 B
Python

from tinygrad.helpers import getenv
import unittest, importlib
@unittest.skipUnless(getenv("MOCKGPU"), 'Testing mockgpu')
class TestMockGPU(unittest.TestCase):
# https://github.com/tinygrad/tinygrad/pull/7627
def test_import_typing_extensions(self):
import test.mockgpu.mockgpu # noqa: F401 # pylint: disable=unused-import
import typing_extensions
importlib.reload(typing_extensions) # pytest imports typing_extension before mockgpu
if __name__ == '__main__':
unittest.main()