def test_summation(): datasets_2d, specs = three_faces_sheet(zaxis=True) datasets = extrude(datasets_2d) eptm = Epithelium('3faces_3D', datasets, specs) eptm_2d = Epithelium('3faces_2D', datasets_2d, specs) edge_copy = datasets['edge'].copy() edge_copy.index = eptm.edge_mindex assert_array_equal(edge_copy.sum(level='cell'), eptm.sum_cell(eptm.edge_df)) eptm_2d.edge_df['test_sum'] = np.linspace(1, eptm_2d.Ne, eptm_2d.Ne) res_sum_srce = eptm_2d.sum_srce(eptm_2d.edge_df['test_sum']) expected_sum_srce = pd.Series([ 21.0, 20.0, 3.0, 4.0, 5.0, 14.0, 9.0, 10.0, 11.0, 26.0, 15.0, 16.0, 17.0 ]) res_sum_trgt = eptm_2d.sum_trgt(eptm_2d.edge_df['test_sum']) expected_sum_trgt = pd.Series([ 36.0, 18.0, 2.0, 3.0, 4.0, 12.0, 8.0, 9.0, 10.0, 24.0, 14.0, 15.0, 16.0 ]) res_sum_face = eptm_2d.sum_face(eptm_2d.edge_df['test_sum']) expected_sum_face = pd.Series([21.0, 57.0, 93.0]) assert (expected_sum_srce == res_sum_srce).all() assert (expected_sum_trgt == res_sum_trgt).all() assert (expected_sum_face == res_sum_face).all()
def test_summation(): datasets_2d, specs = three_faces_sheet(zaxis=True) datasets = extrude(datasets_2d) eptm = Epithelium("3faces_3D", datasets, specs) data = eptm.edge_df.index.values assert_array_equal( eptm.sum_cell(data).values.flatten(), np.array([1278, 1926, 2574])) sum_trgt = np.array([ 462, 302, 88, 97, 106, 248, 142, 151, 160, 356, 196, 205, 214, 501, 340, 107, 116, 125, 286, 161, 170, 179, 394, 215, 224, 233, ]) assert_array_equal(eptm.sum_trgt(data).values.flatten(), sum_trgt) sum_srce = np.array([ 441, 300, 87, 96, 105, 246, 141, 150, 159, 354, 195, 204, 213, 522, 342, 108, 117, 126, 288, 162, 171, 180, 396, 216, 225, 234, ]) assert_array_equal(eptm.sum_srce(data).values.flatten(), sum_srce) sum_face = np.array([ 15, 51, 87, 123, 159, 195, 150, 166, 182, 198, 214, 230, 246, 262, 278, 294, 310, 326, 342, 358, 374, 390, 406, 422, ]) assert_array_equal(eptm.sum_face(data).values.flatten(), sum_face)