github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

setitem tests (some failing) from kernelize (#9940)

Browse Source
This commit is contained in:
qazal
2025-04-20 13:47:55 +03:00
committed by GitHub
parent 6c30948df6
commit 9a9aba4cd5
2 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/unit/test_disk_tensor.py
View File

@@ -266,6 +266,14 @@ class TestDiskTensor(unittest.TestCase):
reloaded = Tensor.empty(10, 10, device=f"disk:{temp('dt_write_ones')}")
np.testing.assert_almost_equal(reloaded.numpy(), np.ones((10, 10)))
def test_simple_setitem(self):
pathlib.Path(temp(fn:="dt_simple_setitem")).unlink(missing_ok=True)
data = [[1],[2]]
src = Tensor(data)
dt = src.to(f"disk:{temp(fn)}")
dt[1] = [3]
self.assertEqual(dt.tolist(), [[1], [3]])
def test_assign_slice(self):
def assign(x,s,y): x[s] = y
helper_test_disk_tensor("dt_assign_slice_1", [0,1,2,3], lambda x: assign(x, slice(0,2), [13, 12]))