Pytest 測試 code (上機考答案)

def calc_tax(e):
sta_mainasu = e - 88000 - 90000 - 128000
if e > 0 and e <= 540000:
return sta_mainasu * 0.05
elif e > 540000 and e <= 1210000:
return sta_mainasu * 0.12 - 37800
elif e > 1210000 and e <= 2420000:
return sta_mainasu * 0.2 - 134600
elif e > 2420000 and e <= 4530000:
return sta_mainasu * 0.3 - 376600
elif e > 4530000 and e <= 10310000:
return sta_mainasu * 0.4 - 829600
elif e > 10310000:
return sta_mainasu * 0.45 - 829600
def test_ansA():
assert calc_tax(540000) == 11700
def test_ansB():
assert calc_tax(630000) == 1080
def test_ansC():
assert calc_tax(1220000) == 48200
def test_ansD():
assert calc_tax(2440000) == 263600
def test_ansE():
assert calc_tax(5630000) == 1300000
def test_ansF():
assert calc_tax(20000000) == 8032700

def Room_Select(people):
if people > 15:
return 1260
elif people > 12:
return 910
elif people > 9:
return 770
elif people > 6:
return 630
elif people > 3:
return 490
elif people > 1:
return 350
else:
return 0
def KTV_Select(People_Count, People_Method):
Method_People = ((350 + 138) * People_Count) * 1.1
Method_Room = (Room_Select(People_Count) * 3 + People_Count * 138) * 1.1
if People_Method == 1:
return (round(Method_People))
else:
return (round(Method_Room))
def test_1():
assert KTV_Select(2, 1) == 1074
def test_2():
assert KTV_Select(3, 1) == 1610
def test_3():
assert KTV_Select(4, 1) == 2147
def test_4():
assert KTV_Select(5, 2) == 2376
def test_5():
assert KTV_Select(8, 2) == 3293
def test_6():
assert KTV_Select(12, 2) == 4363
def test_7():
assert KTV_Select(15, 2) == 5280
def test_8():
assert KTV_Select(20, 2) == 7194