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