25 lines
655 B
Python
25 lines
655 B
Python
from lightphe.commons.logger import Logger
|
|
|
|
logger = Logger(module="tests/test_salary.py")
|
|
|
|
|
|
def test_api():
|
|
from lightphe import LightPHE
|
|
|
|
cs = LightPHE(algorithm_name="Paillier", key_size=50)
|
|
|
|
# set initial base salary to 10000 usd
|
|
salary = 10000
|
|
salary_encrypted = cs.encrypt(salary)
|
|
|
|
# add 1000 usd to base salary
|
|
wage_increase = 1000
|
|
increase_encrypted = cs.encrypt(wage_increase)
|
|
|
|
# perform homomorphic addition
|
|
assert cs.decrypt(salary_encrypted + increase_encrypted) == salary + wage_increase
|
|
|
|
# increase base salary 5%
|
|
ratio = 1.05
|
|
assert cs.decrypt(salary_encrypted * ratio) == salary * ratio
|