mirror of https://github.com/commaai/tinygrad.git
hotfix: add test_tiny, because many times it's what you want
This commit is contained in:
parent
c4db927c7b
commit
b20b22a738
|
@ -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()
|
||||
|
Loading…
Reference in New Issue