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)
