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)
