mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-30 19:33:45 +08:00
github-actions[bot]
eefc084302
version: sunnypilot v2025.003.000 (dev) date: 2025-12-18T05:48:43 master commit: 16c052af0835f049a67b80251d8cc1114fc08bb4
15 lines
428 B
Python
15 lines
428 B
Python
import unittest
|
|
from tinygrad.helpers import GlobalCounters
|
|
from tinygrad.nn.datasets import mnist
|
|
|
|
class TestDataset(unittest.TestCase):
|
|
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()
|