def test_size(self): ndl = led_matrix.NDList(shape=(0, )) self.assertEqual(ndl.size, 0) ndl = led_matrix.NDList(shape=(1, )) self.assertEqual(ndl.size, 1) ndl = led_matrix.NDList(shape=(15, 24, 2)) self.assertEqual(ndl.size, 720)
def test_shape(self): ndl = led_matrix.NDList(shape=(0, )) self.assertEqual(ndl, []) ndl = led_matrix.NDList(shape=(17, )) self.assertEqual(ndl.shape, (17, )) ndl = led_matrix.NDList(shape=(71, 17, 303)) self.assertEqual(ndl.shape, (71, 17, 303))
def test_get_10x10(self): # Test a 10x10 array num = random() * 10 ndl = led_matrix.NDList(shape=(10, 10), fill=num) for i in range(10): for j in range(10): self.assertEqual(ndl[i, j], num)
def test_set_5x5(self): # Test a 5x5 array ndl = led_matrix.NDList(shape=(5, 5)) for i in range(5): for j in range(5): num = random() * 10 ndl[i, j] = num self.assertEqual(ndl[i, j], num)
def test_get_4x4(self): # Test a 4x4x4 array num = random() * 10 ndl = led_matrix.NDList(shape=(5, 5, 5), fill=num) for i in range(5): for j in range(5): for k in range(5): self.assertEqual(ndl[i, j, k], num)
def test_set_2x4(self): # Test array with varrying dimensions ndl = led_matrix.NDList(shape=(2, 4)) for i in range(2): for j in range(4): num = random() * 10 ndl[i, j] = num self.assertEqual(ndl[i, j], num)
def test_set_4x4(self): # Test a 4x4x4 array ndl = led_matrix.NDList(shape=(4, 4, 4)) for i in range(4): for j in range(4): for k in range(4): num = random() * 10 ndl[i, j, k] = num self.assertEqual(ndl[i, j, k], num)
def test_get_17x7x41x90x1(self): # Advanced Test num = random() * 10 ndl = led_matrix.NDList(shape=(17, 7, 41, 90, 1), fill=num) for i in range(17): for j in range(7): for k in range(41): for l in range(90): for m in range(1): self.assertEqual(ndl[i, j, k, l, m], num)
def test_set_17x7x41x1x23x2(self): # Advanced Test ndl = led_matrix.NDList(shape=(17, 7, 41, 1, 23, 2)) for i in range(17): for j in range(7): for k in range(41): for m in range(1): for n in range(23): for o in range(2): num = random() * 10 ndl[i, j, k, m, n, o] = num self.assertEqual(ndl[i, j, k, m, n, o], num)
def test_set_single_embedded(self): # Test a single embedded list ndl = led_matrix.NDList(shape=(1, 1)) ndl[0, 0] = 7 self.assertEqual(ndl[0, 0], 7)
def test_set_single(self): # Test an array with a single element ndl = led_matrix.NDList(shape=(1, )) ndl[0] = 7 self.assertEqual(ndl[0], 7)
def test_init_0(self): ndl = led_matrix.NDList(shape=(0, )) self.assertEqual(ndl, [])
def test_get_single(self): # Test a single embedded list num = random() * 10 ndl = led_matrix.NDList(shape=(1, ), fill=num) self.assertEqual(ndl[0], num)
def test_init_17x7x41x90x1(self): ndl = led_matrix.NDList(shape=(17, 7, 41, 90, 1)) ndl_list = [[[[[0.0 for _ in range(1)] for _ in range(90)] for _ in range(41)] for _ in range(7)] for _ in range(17)] self.assertEqual(ndl, ndl_list)
def test_init_1(self): ndl = led_matrix.NDList(shape=(1, )) self.assertEqual(ndl, [0.0])