ThoughtWorks的代码题 https://www.jinshuju.net/f/EGQL3D

tw.py

def transform(specials, mapping, n):
    """
    https://www.jinshuju.net/f/EGQL3D

    """
    # rule 5
    if str(specials[0]) in str(n):
        return mapping[specials[0]]

    return ''.join(mapping[i] for i in specials if n % i == 0) or str(n)


def test():
    right_answers = """
    1
2
Fizz
4
Buzz
Fizz
Whizz
8
Fizz
Buzz
11
Fizz
Fizz
Whizz
FizzBuzz
16
17
Fizz
19
Buzz
    """

    specials = [3, 5, 7]
    mapping = {3: 'Fizz', 5: 'Buzz', 7: 'Whizz'}

    right_answers = [line.strip() for line in right_answers.split('\n') if line.strip()]
    for i in range(1, len(right_answers)):
        assert transform(specials, mapping, i) == right_answers[i - 1]

    print 'test pass'


def main():
    specials = []
    while len(specials) != 3:
        raw = raw_input('input three less than 10 numbers: ')
        specials = [int(n) for n in raw.split(' ') if n.isdigit() and int(n) < 10]

    mapping = dict(zip(specials, ['Fizz', 'Buzz', 'Whizz']))

    for i in range(1, 100):
        print transform(specials, mapping, i)


if __name__ == "__main__":
    test()
    main()