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