def test_multiply_ndarray(self): r = np.multiply(self.a, self.y) r_enc = np.multiply(self.a_enc, self.y) r_dec = he.decrypt_ndarray(r_enc) r = np.around(r, PRECISION) r_dec = np.around(r_dec, PRECISION) self.assertSequenceEqual(r.tolist(), r_dec.tolist())
def test_sub_ndarray(self): r = self.a - self.y r_enc = self.a_enc - self.y r_dec = he.decrypt_ndarray(r_enc) r = np.around(r, PRECISION) r_dec = np.around(r_dec, PRECISION) self.assertSequenceEqual(r.tolist(), r_dec.tolist())