def test_as_array_is_view(self): b = Matrix(5, 4, 3) a = b.as_array() a[:] = 23. self.assertEqual(b[0], 23.) self.assertEqual(b[5], 23.) self.assertEqual(b[12], 23.)
def test_memory_management1(self): b = Matrix(10) a = b.as_array() del b gc.collect() a[1] = 1 del a gc.collect()
def test_as_array(self): b = Matrix(5, 4, 3) a = b.as_array() self.assertEqual(a.shape, (5, 4, 3)) self.assertTrue(np.all(a == 0.))