mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
d59e6e7a37
* move more tests to test/null, split some existing ones * null work * null work * move more * fixes * move PIL * PIL in CLIP * don't move that
22 lines
585 B
Python
22 lines
585 B
Python
import unittest
|
|
from tinygrad import Tensor
|
|
from tinygrad.nn.state import TensorIO
|
|
|
|
class TestTensorIO(unittest.TestCase):
|
|
def test_read(self):
|
|
data = b"Hello World!"
|
|
fobj = TensorIO(Tensor(data))
|
|
self.assertEqual(fobj.read(1), data[:1])
|
|
self.assertEqual(fobj.read(5), data[1:6])
|
|
self.assertEqual(fobj.read(100), data[6:])
|
|
self.assertEqual(fobj.read(100), b"")
|
|
|
|
def test_read_nolen(self):
|
|
data = b"Hello World!"
|
|
fobj = TensorIO(Tensor(data))
|
|
fobj.seek(2)
|
|
self.assertEqual(fobj.read(), data[2:])
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|