def test_nonorthogonal_transform_round_trip(self):
transform = NonOrthogonalTransform(angle=np.radians(40.))
x, y = np.array([1., 2., 3.]), np.array([4., 5., 6.])
xp, yp = transform.tr(x, y)
xpinv, ypinv = transform.inv_tr(xp, yp)
assert_allclose(x, xpinv)
assert_allclose(y, ypinv)
def test_nonorthogonal_transform_skews_as_expected(self):
transform = NonOrthogonalTransform(angle=np.radians(45.)) # 45deg
x = np.array([0, 1])
y = np.array([1, 0])
xp, yp = transform.tr(x, y)
assert_allclose(xp, np.array([1. / np.sqrt(2.), 1.]))
assert_allclose(yp, np.array([1. / np.sqrt(2.), 0.]))
def test_create_nonorthogonal_transform_with_orthogonal_lattice_and_nonorthogonal_projections(
self):
lattice = OrientedLattice() # default angle=90 degrees
transform = NonOrthogonalTransform.from_lattice(lattice,
x_proj=(1, 0, 0),
y_proj=(1, 1, 0))
self.assertAlmostEqual(transform.angle, 0.25 * np.pi)
def test_create_nonorthogonal_transform_with_nonorthogonal_lattice_and_nonorthogonal_projections(
self):
lattice = OrientedLattice(1, 1, 2, 90, 90, 120.)
x_proj, y_proj = (1, 0, 0), (1, 1, 0)
transform = NonOrthogonalTransform.from_lattice(
lattice, x_proj, y_proj)
self.assertAlmostEqual(transform.angle,
np.radians(lattice.recAngle(*x_proj, *y_proj)))
def test_set_transform(self):
# make sliceinfo with unit transform (default)
info = SliceInfo(frame=SpecialCoordinateSystem.HKL,
point=(None, None, 0.5),
transpose=False,
range=[None, None, (-15, 15)],
qflags=[True, True, True])
info.set_transform(NonOrthogonalTransform(angle=radians(120)))
self.assertEqual(info.can_support_nonorthogonal_axes(), True)
def setUp(self):
self.image = MagicMock()
self.nonortho_transform = NonOrthogonalTransform(radians(60))
self.image_info_widget = MagicMock(ImageInfoWidget)