def test_delete_wrong_type(self): r = Matrix(self.data) r.delete_columns("hi") # bang, cannot delete columns named as string
def test_delete_columns(self): r = Matrix(self.data) # delete a list of indices r.delete_columns([0, 2]) assert r.row[0] == [2, 4, 5, 6]