def test_compute_centroid(): # ------------------------------------------------------------------------ # tests computed centroid is same as MouseConnApi with 'true' inj density # pull 'data' from mcc fixture a = np.random.rand(4, 4, 4) b = np.random.rand(4, 4, 4) # compute allensdk centroid api = MouseConnectivityApi() mcc_centroid = api.calculate_injection_centroid(a, b, 1) # 'true' injection density _compute_true_injection_density(a, b, inplace=True) assert_array_almost_equal(compute_centroid(a), mcc_centroid)