def test_comap_copy(): incs, outcs, map, inv = voxel_to_world() cm = CoordinateMap(map, incs, outcs, inv) cmcp = cm.copy() yield assert_equal, cmcp.mapping, cm.mapping yield assert_equal, cmcp.input_coords, cm.input_coords yield assert_equal, cmcp.output_coords, cm.output_coords yield assert_equal, cmcp.inverse_mapping, cm.inverse_mapping
def test_reordered_range(): incs, outcs, map, inv = voxel_to_world() cm = CoordinateMap(incs, outcs, inv, map) recm = cm.reordered_range('yzx') yield assert_equal, recm.function_domain.coord_names, incs.coord_names yield assert_equal, recm.function_range.coord_names, ('y', 'z', 'x') yield assert_equal, recm.function_domain.name, incs.name yield assert_equal, recm.function_range.name, outcs.name # default reverse order recm = cm.reordered_range() yield assert_equal, recm.function_range.coord_names, ('z', 'y', 'x') # reorder with indicies recm = cm.reordered_range([2,0,1]) yield assert_equal, recm.function_range.coord_names, ('z', 'x', 'y')
def test_reordered_domain(): incs, outcs, map, inv = voxel_to_world() cm = CoordinateMap(incs, outcs, map, inv) recm = cm.reordered_domain('jki') yield assert_equal, recm.function_domain.coord_names, ('j', 'k', 'i') yield assert_equal, recm.function_range.coord_names, outcs.coord_names yield assert_equal, recm.function_domain.name, incs.name yield assert_equal, recm.function_range.name, outcs.name # default reverse reorder recm = cm.reordered_domain() yield assert_equal, recm.function_domain.coord_names, ('k', 'j', 'i') # reorder with order as indices recm = cm.reordered_domain([2,0,1]) yield assert_equal, recm.function_domain.coord_names, ('k', 'i', 'j')