def test_sum_multiple(self): data = np.array([list(range(0, 20))] * 3) expected_values = np.sum(data, axis=1) self.set_seed() value = DiffPrivStatistics.sum(data, self.epsilon, axis=1) np.testing.assert_almost_equal(value, expected_values, self.decimal_places)
def test_sum_single(self): data = np.array(list(range(0, 20))) expected_value = np.sum(data) self.set_seed() value = DiffPrivStatistics.sum(data, self.epsilon) self.assertAlmostEqual(value, expected_value, self.decimal_places)