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 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()
