def test_mean_multiple(self): data = np.array([list(range(0, 20)) + [100.0]] * 3) expected_values = np.mean(data, axis=1) self.set_seed() value = DiffPrivStatistics.mean(data, self.epsilon, axis=1) np.testing.assert_almost_equal(value, expected_values, self.decimal_places)
def test_mean_single(self): data = np.array(list(range(0, 20)) + [100.0]) expected_value = np.mean(data) self.set_seed() value = DiffPrivStatistics.mean(data, self.epsilon) self.assertAlmostEqual(value, expected_value, self.decimal_places)