def test_correct_month_input(self): import sys sys.argv = ['', 'csv', 'asdf', '3'] months = getMonths() self.assertEqual(months, float('3'))
def test_no_month_input(self): import sys sys.argv = ['', 'csv', 'borrowing'] month = getMonths() self.assertEqual(month, 12)