You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
456 B
16 lines
456 B
1 week ago
|
import unittest
|
||
|
|
from tinygrad.helpers import GlobalCounters
|
||
|
|
from tinygrad.nn.datasets import mnist
|
||
|
|
|
||
|
|
class TestDataset(unittest.TestCase):
|
||
|
|
@unittest.expectedFailure
|
||
|
|
def test_dataset_is_realized(self):
|
||
|
|
X_train, _, _, _ = mnist()
|
||
|
|
X_train[0].contiguous().realize()
|
||
|
|
start = GlobalCounters.kernel_count
|
||
|
|
X_train[0].contiguous().realize()
|
||
|
|
self.assertEqual(GlobalCounters.kernel_count-start, 1)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|