def test_remove(self): m = util.Matrix(1, 2) m[0, 0] = 1 m[0, 1] = 2 m.remove(2) self.assertEquals(m[0, 0], 1) self.assertEquals(m[0, 1], None)
def test_rows(self): m = util.Matrix(3, 2) m[0, 0] = 1 m[0, 1] = 2 m[1, 0] = 3 m[1, 1] = 4 m[2, 0] = 5 m[2, 1] = 6 self.assertEquals(list(m.row(0)), [1, 3, 5]) self.assertEquals(list(m.row(1)), [2, 4, 6])
def test_columns(self): m = util.Matrix(3, 2) m[0, 0] = 1 m[0, 1] = 2 m[1, 0] = 3 m[1, 1] = 4 m[2, 0] = 5 m[2, 1] = 6 self.assertEquals(list(m.column(0)), [1, 2]) self.assertEquals(list(m.column(1)), [3, 4]) self.assertEquals(list(m.column(2)), [5, 6])
def test_get_set(self): m = util.Matrix(3, 2) m[0, 0] = 1 m[0, 1] = 2 m[1, 0] = 3 m[1, 1] = 4 m[2, 0] = 5 m[2, 1] = 6 self.assertEquals(m[0, 0], 1) self.assertEquals(m[1, 0], 3) self.assertEquals(m[2, 0], 5) m[0, 0] = 17 self.assertEquals(m[0, 0], 17)
def test_matrix_init(self): m = util.Matrix(1, 2) self.assertEquals(list(m), [None, None]) m = util.Matrix(2, 1) self.assertEquals(list(m), [None, None]) m = util.Matrix(1, 5) self.assertEquals(m.columns, 1) self.assertEquals(m.rows, 5) m = util.Matrix(0, 0) self.assertEquals(m.columns, 0) self.assertEquals(m.rows, 0) m = util.Matrix(5, 1) self.assertEquals(m.columns, 5) self.assertEquals(m.rows, 1) self.assertEquals(m[0, 0], None) self.assertEquals(m[1, 0], None) self.assertEquals(m[2, 0], None) self.assertEquals(m[3, 0], None) self.assertEquals(m[4, 0], None)