def test_upcast(): datasets_2d, specs = three_faces_sheet(zaxis=True) datasets = extrude(datasets_2d) eptm = Epithelium("3faces_3D", datasets, specs) eptm.cell_df["test_data"] = eptm.cell_df.index eptm.face_df["test_data"] = eptm.face_df.index eptm.vert_df["test_data"] = eptm.vert_df.index assert_array_equal(eptm.upcast_srce(eptm.vert_df["test_data"]), eptm.edge_df["srce"]) assert_array_equal(eptm.upcast_srce("test_data"), eptm.edge_df["srce"]) assert_array_equal(eptm.upcast_trgt(eptm.vert_df["test_data"]), eptm.edge_df["trgt"]) assert_array_equal(eptm.upcast_trgt("test_data"), eptm.edge_df["trgt"]) assert_array_equal(eptm.upcast_face("test_data"), eptm.edge_df["face"]) assert_array_equal(eptm.upcast_cell("test_data"), eptm.edge_df["cell"])
def test_upcast_ndarray(): datasets_2d, specs = three_faces_sheet(zaxis=True) datasets = extrude(datasets_2d) eptm = Epithelium("3faces_3D", datasets, specs) data = np.arange(eptm.Nv * 3).reshape((eptm.Nv, 3)) assert eptm.upcast_srce(data).shape == (eptm.Ne, 3)