class Test__apply_axes_mapping(unittest.TestCase): def setUp(self): self.zeros = ConstantArray([10, 20, 40, 50]) self.a = TransposedArray(self.zeros, [1, 3, 0, 2]) def test_too_few_dims(self): msg = 'length 3, but should be of length 4' with self.assertRaisesRegexp(ValueError, msg): self.a._apply_axes_mapping(list('abc')) def test_correct_n_dims(self): r = self.a._apply_axes_mapping(list('abcd')) self.assertEqual(r, ('b', 'd', 'a', 'c')) def test_inverse(self): r = self.a._apply_axes_mapping(list('bdac'), inverse=True) self.assertEqual(r, tuple('abcd')) def test_too_many_dims(self): msg = 'length 5, but should be of length 4' with self.assertRaisesRegexp(ValueError, msg): self.a._apply_axes_mapping(list('abcde'))
class Test__apply_axes_mapping(unittest.TestCase): def setUp(self): self.zeros = ConstantArray([10, 20, 40, 50]) self.a = TransposedArray(self.zeros, [1, 3, 0, 2]) def test_too_few_dims(self): msg = 'length 3, but should be of length 4' with six.assertRaisesRegex(self, ValueError, msg): self.a._apply_axes_mapping(list('abc')) def test_correct_n_dims(self): r = self.a._apply_axes_mapping(list('abcd')) self.assertEqual(r, ('b', 'd', 'a', 'c')) def test_inverse(self): r = self.a._apply_axes_mapping(list('bdac'), inverse=True) self.assertEqual(r, tuple('abcd')) def test_too_many_dims(self): msg = 'length 5, but should be of length 4' with six.assertRaisesRegex(self, ValueError, msg): self.a._apply_axes_mapping(list('abcde'))