def testCanCreateSniffedReaderForEmptyData(self): emptyReadable = StringIO.StringIO("") self.assertTrue(sniff.delimitedOptions(emptyReadable)) emptyDataFormat = sniff.createDataFormat(emptyReadable) self.assertTrue(emptyDataFormat) self.assertEqual(emptyDataFormat.name, data.FORMAT_DELIMITED) emptyReader = sniff.createReader(emptyReadable) self.assertTrue(emptyReadable) rowCount = 0 for _ in emptyReader: rowCount += 1 self.assertEqual(rowCount, 0)
def testCanFindOutSniffDataFormat(self): fileNameToExpectedFormatMap = { "valid_customers.csv": data.FORMAT_DELIMITED, "valid_customers.ods": data.FORMAT_ODS, "valid_customers.xls": data.FORMAT_EXCEL, } for testFileName, exptectedDataFormatName in fileNameToExpectedFormatMap.items( ): testFilePath = dev_test.getTestInputPath(testFileName) testFile = open(testFilePath, "rb") try: actualDataFormat = sniff.createDataFormat(testFile) actualDataFormatName = actualDataFormat.name self.assertEqual( actualDataFormatName, exptectedDataFormatName, "data format for file must be %r but is %r: %r" % (exptectedDataFormatName, actualDataFormatName, testFilePath)) finally: testFile.close()
def testCanFindOutSniffDataFormat(self): fileNameToExpectedFormatMap = { "valid_customers.csv": data.FORMAT_DELIMITED, "valid_customers.ods": data.FORMAT_ODS, "valid_customers.xls": data.FORMAT_EXCEL, } for testFileName, exptectedDataFormatName in fileNameToExpectedFormatMap.items(): testFilePath = dev_test.getTestInputPath(testFileName) testFile = open(testFilePath, "rb") try: actualDataFormat = sniff.createDataFormat(testFile) actualDataFormatName = actualDataFormat.name self.assertEqual( actualDataFormatName, exptectedDataFormatName, "data format for file must be %r but is %r: %r" % (exptectedDataFormatName, actualDataFormatName, testFilePath), ) finally: testFile.close()