codeeval fizzbuzz

fizzbuzz.py

# coding=utf-8

import sys


def fizzbuzz(a, b, n):
    tmp = []

    if n % a == 0:
        tmp.append("F")
    if n % b == 0:
        tmp.append("B")
    if len(tmp) == 0:
        tmp.append(str(n))
    return "".join(tmp)


test_cases = open(sys.argv[1], 'r')
for line in test_cases:
    a, b, n = [int(e) for e in line.split()]
    print " ".join([fizzbuzz(a, b, i) for i in range(1, n + 1)])

test_cases.close()
test_cases.close()