def test_coord_transform(self): cyl_coords_1 = (1.0,1.0,1.0) dummy_coords_1 = Coords('cylindrical',cyl_coords_1) dummy_coords_1.load_transform(DummyCoordTransform) # The DummyCoordinateTransform should map (a,b,c) -> (2a,3b,4c) self.assertEqual(dummy_coords_1.dummy(), (2*cyl_coords_1[0], 3*cyl_coords_1[1], 4*cyl_coords_1[2])) # Check again with different ccoordinated. cyl_coords_2 = (2.0,1.0,4.0) dummy_coords_2 = Coords('cylindrical',cyl_coords_2) dummy_coords_2.load_transform(DummyCoordTransform) self.assertEqual(dummy_coords_2.dummy(), (2*cyl_coords_2[0], 3*cyl_coords_2[1], 4*cyl_coords_2[2]))