题目是这样的:
实现对一个四位整数的加密过程,返回加密数。加密方法:对该四位数的每一位都加5,并用和对10取余代替该数字,在将第一位与第四位交换,第二位与第三位交换,得到一个加密数。
下面是实现方法
新建encrypt.py
文件:
# coding:utf-8
# author: Allen.Z
# date: 2020-09-02
# 实现对一个四位整数的加密过程,返回加密数:
# 对该四位数的每一位都加5,并用和对10取余代替该数字,
# 在将第一位与第四位交换,第二位与第三位交换,得到一个加密数
def encrypt(num):
"""
加密一个四位数字的算法
"""
s = str(num)
a = (int(s[3]) + 5) % 10
b = (int(s[1]) + 5) % 10
c = (int(s[2]) + 5) % 10
d = (int(s[0]) + 5) % 10
return a * 1000 + b * 100 + c * 10 + d
添加如下测试代码:
def test_encrypt():
numbers = [
(1234, 9786),
(8399, 4843),
(1535, 86),
]
for n in numbers:
assert encrypt(n[0]) == n[1]
测试一下:
完美!