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)
