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