def test_transpose_dims(self): xdata = numpy.array([[1.0, 2.0], [3.0, 4.0]], dtype="d") X = PhysArray(xdata, name="X", units="m", dimensions=("u", "v")) indata = ("v", "u") testname = "X.transpose({}, {})".format(*indata) actual = X.transpose(*indata) new_name = "transpose({}, from=[u,v], to=[v,u])".format(X.name) expected = PhysArray([[1.0, 3.0], [2.0, 4.0]], units=X.units, name=new_name, dimensions=indata) print_test_message(testname, actual=actual, expected=expected, X=X) self.assertPhysArraysEqual(actual, expected, testname=testname)
def test_transpose_dims(self): xdata = numpy.array([[1., 2.], [3., 4.]], dtype='d') X = PhysArray(xdata, name='X', units='m', dimensions=('u', 'v')) indata = ('v', 'u') testname = 'X.transpose({}, {})'.format(*indata) actual = X.transpose(*indata) new_name = "transpose({}, from=[u,v], to=[v,u])".format(X.name, indata) expected = PhysArray([[1., 3.], [2., 4.]], units=X.units, name=new_name, dimensions=indata) print_test_message(testname, actual=actual, expected=expected, X=X) self.assertPhysArraysEqual(actual, expected, testname=testname)