def test_secint(self):
secint = mpc.SecInt()
y = [1, 3, -2, 3, 1, -2, -2, 4] * 5
random.shuffle(y)
x = list(map(secint, y))
self.assertEqual(mpc.run(mpc.output(mean(x))), round(statistics.mean(y)))
self.assertEqual(mpc.run(mpc.output(variance(x))), round(statistics.variance(y)))
self.assertEqual(mpc.run(mpc.output(variance(x, mean(x)))), round(statistics.variance(y)))
self.assertEqual(mpc.run(mpc.output(stdev(x))), round(statistics.stdev(y)))
self.assertEqual(mpc.run(mpc.output(pvariance(x))), round(statistics.pvariance(y)))
self.assertEqual(mpc.run(mpc.output(pstdev(x))), round(statistics.pstdev(y)))
self.assertEqual(mpc.run(mpc.output(mode(x))), round(statistics.mode(y)))
self.assertEqual(mpc.run(mpc.output(median(x))), round(statistics.median(y)))
self.assertEqual(mpc.run(mpc.output(median_low(x))), round(statistics.median_low(y)))
self.assertEqual(mpc.run(mpc.output(median_high(x))), round(statistics.median_high(y)))
self.assertEqual(mpc.run(mpc.output(quantiles(x[:2], n=3))),
statistics.quantiles(y[:2], n=3))
self.assertEqual(mpc.run(mpc.output(quantiles(x, n=1))), statistics.quantiles(y, n=1))
self.assertEqual(mpc.run(mpc.output(quantiles(x))), statistics.quantiles(y))
x = list(range(16))
y = list(reversed(x))
self.assertAlmostEqual(covariance(x, y), -22.667, 3)
x = list(map(secint, x))
y = list(map(secint, y))
self.assertEqual(mpc.run(mpc.output(covariance(x, y))), -23)
self.assertRaises(ValueError, quantiles, x, method='wrong')
def test_secint(self):
secint = mpc.SecInt()
y = [1, 3, -2, 3, 1, -2, -2, 4] * 5
random.shuffle(y)
x = list(map(secint, y))
self.assertEqual(mpc.run(mpc.output(mean(x))),
round(statistics.mean(y)))
self.assertEqual(mpc.run(mpc.output(variance(x))),
round(statistics.variance(y)))
self.assertEqual(mpc.run(mpc.output(variance(x, mean(x)))),
round(statistics.variance(y)))
self.assertEqual(mpc.run(mpc.output(stdev(x))),
round(statistics.stdev(y)))
self.assertEqual(mpc.run(mpc.output(pvariance(x))),
round(statistics.pvariance(y)))
self.assertEqual(mpc.run(mpc.output(pstdev(x))),
round(statistics.pstdev(y)))
self.assertEqual(mpc.run(mpc.output(mode(x))),
round(statistics.mode(y)))
self.assertEqual(mpc.run(mpc.output(median(x))),
round(statistics.median(y)))
self.assertEqual(mpc.run(mpc.output(median_low(x))),
round(statistics.median_low(y)))
self.assertEqual(mpc.run(mpc.output(median_high(x))),
round(statistics.median_high(y)))
def test_secfxp(self):
secfxp = mpc.SecFxp()
x = [1, 1, 2, 2, 3, 4, 4, 4, 6] * 5
random.shuffle(x)
x = list(map(secfxp, x))
self.assertAlmostEqual(mpc.run(mpc.output(mean(x))), 3, delta=1)
self.assertAlmostEqual(mpc.run(mpc.output(median(x))), 3)
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))), 4)
x = [1, 1, 1, 1, 2, 2, 3, 4, 4, 4, 4, 5, 6, 6, 6] * 100
random.shuffle(x)
x = list(map(lambda a: a * 2**-4, x))
x = list(map(secfxp, x))
self.assertAlmostEqual(mpc.run(mpc.output(mean(x))), (2**-4) * 10/3, delta=1)
y = [1.75, 1.25, -0.25, 0.5, 1.25, -3.5] * 5
random.shuffle(y)
x = list(map(secfxp, y))
self.assertAlmostEqual(mpc.run(mpc.output(mean(x))), statistics.mean(y), 4)
self.assertAlmostEqual(mpc.run(mpc.output(variance(x))), statistics.variance(y), 2)
self.assertAlmostEqual(mpc.run(mpc.output(stdev(x))), statistics.stdev(y), 3)
self.assertAlmostEqual(mpc.run(mpc.output(pvariance(x))), statistics.pvariance(y), 2)
self.assertAlmostEqual(mpc.run(mpc.output(pstdev(x))), statistics.pstdev(y), 3)
self.assertAlmostEqual(mpc.run(mpc.output(median(x))), statistics.median(y), 4)
self.assertAlmostEqual(mpc.run(mpc.output(quantiles(x))), statistics.quantiles(y), 4)
self.assertAlmostEqual(mpc.run(mpc.output(quantiles(x, method='inclusive'))),
statistics.quantiles(y, method='inclusive'), 4)
x = list(map(secfxp, [1.0]*10))
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))), 1)
k = mpc.options.sec_param
mpc.options.sec_param = 1 # force no privacy case
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))), 1)
mpc.options.sec_param = k
x[0] = secfxp(1.5)
self.assertRaises(ValueError, mode, x)
x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [1, 2, 3, 1, 2, 3, 1, 2, 3]
self.assertEqual(covariance(x, y), 0.75)
self.assertEqual(correlation(x, x), 1.0)
self.assertAlmostEqual(correlation(x, y), 0.316, 3)
self.assertEqual(linear_regression(x, y)[1], 1.5)
x = list(map(secfxp, x))
y = list(map(secfxp, y))
self.assertEqual(mpc.run(mpc.output(covariance(x, y))), 0.75)
self.assertAlmostEqual(mpc.run(mpc.output(correlation(x, x))), 1.0, 2)
self.assertAlmostEqual(mpc.run(mpc.output(correlation(x, y))), 0.32, 2)
self.assertAlmostEqual(mpc.run(mpc.output(linear_regression(x, y)[1])), 1.5, 2)
x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [9, 8, 7, 6, 5, 4, 3, 2, 1]
self.assertEqual(covariance(x, y), -7.5)
self.assertEqual(correlation(x, y), -1.0)
self.assertEqual(linear_regression(x, y)[1], 10.0)
x = list(map(secfxp, x))
y = list(map(secfxp, y))
self.assertAlmostEqual(mpc.run(mpc.output(covariance(x, y))), -7.5, 2)
self.assertAlmostEqual(mpc.run(mpc.output(correlation(x, y))), -1.0, 2)
self.assertAlmostEqual(mpc.run(mpc.output(linear_regression(x, y)[1])), 10.0, 2)
def test_plain(self):
f = lambda: (i * j for i in range(-1, 2, 1) for j in range(2, -2, -1))
self.assertEqual(mean(f()), statistics.mean(f()))
self.assertEqual(variance(f()), statistics.variance(f()))
self.assertEqual(stdev(f()), statistics.stdev(f()))
self.assertEqual(pvariance(f()), statistics.pvariance(f()))
self.assertEqual(pstdev(f()), statistics.pstdev(f()))
self.assertEqual(mode(f()), statistics.mode(f()))
self.assertEqual(median(f()), statistics.median(f()))
def test_plain(self):
f = lambda: (i * j for i in range(-1, 2, 1) for j in range(2, -2, -1))
self.assertEqual(mean(f()), statistics.mean(f()))
self.assertEqual(variance(f()), statistics.variance(f()))
self.assertEqual(stdev(f()), statistics.stdev(f()))
self.assertEqual(pvariance(f()), statistics.pvariance(f()))
self.assertEqual(pstdev(f()), statistics.pstdev(f()))
self.assertEqual(mode(f()), statistics.mode(f()))
self.assertEqual(median(f()), statistics.median(f()))
self.assertEqual(quantiles(f()), statistics.quantiles(f()))
self.assertEqual(quantiles(f(), n=6, method='inclusive'),
statistics.quantiles(f(), n=6, method='inclusive'))
x = list(f())
y = list(reversed(x))
self.assertEqual(covariance(x, y), statistics.covariance(x, y))
self.assertEqual(correlation(x, y), statistics.correlation(x, y))
self.assertEqual(linear_regression(x, y), statistics.linear_regression(x, y))
def test_secfxp(self):
secfxp = mpc.SecFxp()
x = [1, 1, 2, 2, 3, 4, 4, 4, 6] * 5
random.shuffle(x)
x = list(map(secfxp, x))
self.assertAlmostEqual(mpc.run(mpc.output(mean(x))).signed(),
3,
delta=1)
self.assertAlmostEqual(mpc.run(mpc.output(median(x))).signed(), 3)
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))).signed(), 4)
x = [1, 1, 1, 1, 2, 2, 3, 4, 4, 4, 4, 5, 6, 6, 6] * 100
random.shuffle(x)
x = list(map(lambda a: a * 2**-4, x))
x = list(map(secfxp, x))
self.assertAlmostEqual(mpc.run(mpc.output(mean(x))).signed(),
(2**-4) * 10 / 3,
delta=1)
y = [1.75, 1.25, -0.25, 0.5, 1.25, -3.5] * 5
random.shuffle(y)
x = list(map(secfxp, y))
self.assertAlmostEqual(float(mpc.run(mpc.output(mean(x)))),
statistics.mean(y), 4)
self.assertAlmostEqual(float(mpc.run(mpc.output(variance(x)))),
statistics.variance(y), 2)
self.assertAlmostEqual(float(mpc.run(mpc.output(stdev(x)))),
statistics.stdev(y), 3)
self.assertAlmostEqual(float(mpc.run(mpc.output(pvariance(x)))),
statistics.pvariance(y), 2)
self.assertAlmostEqual(float(mpc.run(mpc.output(pstdev(x)))),
statistics.pstdev(y), 3)
self.assertAlmostEqual(float(mpc.run(mpc.output(median(x)))),
statistics.median(y), 4)
x = list(map(secfxp, [1.0] * 10))
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))).signed(), 1)
k = mpc.options.sec_param
mpc.options.sec_param = 1 # force no privacy case
self.assertAlmostEqual(mpc.run(mpc.output(mode(x))).signed(), 1)
mpc.options.sec_param = k