def test_row_change(self): ar0 = [['a', 'b', 'c'], ['a', 'd', 'g']] ar1_a = [['a', 'a', 'a'], ['a', 'a', 'a']] ar1_b = [['b', 'b', 'b'], ['d', 'd', 'd']] ar1_c = [['c', 'b', 'a'], ['g', 'd', 'a']] array0 = SymArray(ar0) new_array_a = array0.switch_cols([0, 0, 0]) new_array_b = array0.switch_cols([1, 1, 1]) new_array_c = array0.switch_cols([2, 1, 0]) self.assertEqual(new_array_a, SymArray(ar1_a)) self.assertEqual(new_array_b, SymArray(ar1_b)) self.assertEqual(new_array_c, SymArray(ar1_c))
def test_rowcols_change(self): ar0 = [['a', 'b', 'c'], ['a', 'd', 'g'], ['b', 'd', 'f']] ar1_a = [['a', 'a', 'a'], ['a', 'a', 'a'], ['a', 'a', 'a']] ar1_b = [['d', 'b', 'f'], ['d', 'a', 'g'], ['b', 'a', 'c']] array0 = SymArray(ar0) # initial array for operations # Checking with ar1_a new_array_a = array0.switch_cols([0, 0, 0]) new_array_a = new_array_a.switch_rows([1, 1, 1]) self.assertEqual(new_array_a, SymArray(ar1_a)) # Checking with ar1_b new_array_b = array0.switch_rows([2, 1, 0]) new_array_b = new_array_b.switch_cols([1, 0, 2]) self.assertEqual(new_array_b, SymArray(ar1_b))