hotfix: add test_tiny, because many times it's what you want

This commit is contained in:
George Hotz 2024-10-14 16:32:33 +08:00
parent c4db927c7b
commit b20b22a738
1 changed files with 11 additions and 0 deletions

11
test/test_tiny.py Normal file
View File

@ -0,0 +1,11 @@
import unittest
from tinygrad import Tensor
class TestTiny(unittest.TestCase):
def test_plus(self):
out = Tensor([1.,2,3]) + Tensor([4.,5,6])
self.assertListEqual(out.tolist(), [5.0, 7.0, 9.0])
if __name__ == '__main__':
unittest.main()