mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-09 22:45:44 +08:00
10 lines
220 B
Python
10 lines
220 B
Python
import unittest
|
|
from tinygrad import Device
|
|
|
|
class TestDeviceCount(unittest.TestCase):
|
|
def test_count(self):
|
|
self.assertGreaterEqual(Device[Device.DEFAULT].count(), 1)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|