def need_update_revenue_tests(): """ test cases """ print('need_update_revenue_tests:') # test for file that does not exist filename = 'data/2023/revenue.csv' if os.path.exists(filename): os.remove(filename) need_update_revenue_test(filename, True) # test for modtime date is today filename = 'data/2023/revenue.csv' fetch.update_revenue('2023', filename) need_update_revenue_test(filename, False) os.remove(filename)
def update_revenue_test(stock, filename, expect): """ test module """ if fetch.update_revenue(stock, filename) == expect: print('pass') else: print('fail')