forked from IQ.Lvbs/IQ.Pilot
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()
|