python 趣味题-计算奖金

题目：企业发放的奖金根据利润提成。利润(I)低于或等于10万元时，奖金可提10%；
利润高于10万元，低于20万元时，低于10万元的部分按10%提成，高于10万元的部分
，可提成7.5%；20万到40万之间时，高于20万元的部分，可提成5%；40万到60万
之间时高于40万元的部分，可提成3%；60万到100万之间时，高于60万元的部分，可提
成1.5%，高于100万元时，超过100万元的部分按1%提成，从键盘输入当月利润I，求
应发放奖金总数？

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2019-07-02 11:17
# @Author  : Allen.Z
# @Desc    : 
# @File    : bonus
"""
题目：企业发放的奖金根据利润提成。利润(I)低于或等于10万元时，奖金可提10%；
利润高于10万元，低于20万元时，低于10万元的部分按10%提成，高于10万元的部分
，可提成7.5%；20万到40万之间时，高于20万元的部分，可提成5%；40万到60万
之间时高于40万元的部分，可提成3%；60万到100万之间时，高于60万元的部分，可提
成1.5%，高于100万元时，超过100万元的部分按1%提成，从键盘输入当月利润I，求
应发放奖金总数？

"""


def bonus(profits):
    amount = 0
    # 提成比例定义
    bns = (0.1, 0.075, 0.05, 0.03, 0.015, 0.01)
    # 提成额度计算
    pfs = [b * 10000 for b in (10, 10, 20, 20, 40)]
    if not profits:
        raise ValueError
    for p, k in zip(pfs, bns):  # 不同额度计算不同的提成金额
        profits = profits - p
        rate = p if profits > 0 else profits + p
        amount = amount + rate * k
        if profits < 0:
            break

    if profits > 0:  # 计算超出100万的提成
        amount = amount + profits * bns[-1]
    return amount


def test_bonus():
    assert bonus(90000) == 9000.0
    b = bonus(1200000)
    assert bonus(120000) == 11500.0
    assert bonus(310000) == 23000.0
    assert bonus(450000) == 29000.0
    assert bonus(800000) == 36500.0
    e = 100000 * 0.1 + 100000 * 0.075 + 200000 * (0.05 + 0.03) + 400000 * 0.015 + 200000 * 0.01
    assert b == e
    print(b, e)