def test_matrix(self): matrix = bspump.Matrix(app=self.App, dtype=[ ('f1', 'i8'), ('f2', 'i8'), ('f3', 'i8'), ]) for i in range(100): n = matrix.add_row() matrix.Array[n][0] = 1 # Access by a field index matrix.Array[n]['f2'] = 1 # Access by a field name matrix.Array[n][2] = 1 self.assertEqual(n, i) closed = set() closed |= matrix.ClosedRows for i in range(20, 40): matrix.close_row(i) closed.add(i) self.assertIn(i, matrix.ClosedRows) self.assertEqual(closed, matrix.ClosedRows) for i in range(20): n = matrix.add_row() self.assertIn(n, closed)
def test_matrix_zeros(self): matrix = bspump.Matrix(app=self.App) for i in range(100): n = matrix.add_row() self.assertEqual(n, i) matrix.zeros() self.assertEqual(matrix.Array.shape, (0, ))
def test_matrix_dtypes(self): matrix = bspump.Matrix(app=self.App, dtype=[ ('f1', 'U20'), ('f2', 'i8'), ('f3', 'i8'), ]) row_index = matrix.add_row() row = matrix.Array[row_index] row['f1'] = "Ahoj" row['f2'] = 64
def test_matrix_flush(self): matrix = bspump.Matrix(app=self.App) for i in range(100): n = matrix.add_row() self.assertEqual(n, i) for i in range(20, 40): matrix.close_row(i) self.assertIn(i, matrix.ClosedRows) matrix.flush() self.assertEqual(len(matrix.ClosedRows), 0)
def test_matrix(self): matrix = bspump.Matrix(app=self.App, dtype=[ ('f1', 'i8'), ('f2', 'i8'), ('f3', 'i8'), ]) for i in range(100): n = matrix.add_row() matrix.Array[n][0] = 1 # Access by a field index matrix.Array[n]['f2'] = 1 # Access by a field name matrix.Array[n][2] = 1 self.assertEqual(n, i)