def testShouldConvertVisaFileWithRange(self): dkb2homebank.convert_visa('testfiles/visaRange.csv', 'visaHomebank.csv') lineNumber = sum(1 for line in open('visaHomebank.csv')) self.assertEqual(lineNumber, 1)
def testShouldConvertVisaFileWithRange(self): with open('testfiles/visaRange.csv', encoding='iso-8859-1') as csv_file: dkb2homebank.convert_visa(csv_file, 'visaHomebank.csv') csv_file.seek(0) lineNumber = len(dkb2homebank.find_transaction_lines(csv_file)) self.assertEqual(lineNumber, 1)