def test_drop_by_condition_filter_and(self):
records = [
r for r in DropByConditionFilter(
self.reader,
condition=IsEqualCondition(2008),
fieldNames=['year', 'after_year']).initialize().getRecords()
]
self.assertEqual(len(records), 58)
self.assertEqual(records[-1].getField('name').getValue(),
'Western Ogre')
def test_logger_filter_default(self):
filtr = DropByConditionFilter(self.reader,
condition=IsEqualCondition(2008),
fieldNames='year')
filtr.setLogFile(logFile='tests/logger', appendLog=False, logger=None)
filtr.initialize()
records = [r for r in filtr.getRecords()]
filtr.finalize()
hashcode = hashlib.md5(open('tests/logger', 'rb').read()).hexdigest()
os.unlink('tests/logger')
self.assertEqual(hashcode, '95eff99740e2d8d2b50808357e006293')
def test_logger_filter_fn(self):
filtr = DropByConditionFilter(self.reader,
condition=IsEqualCondition(2008),
fieldNames='year')
filtr.setLogFile(logFile='tests/logger',
appendLog=False,
logger='tests.test_filter.notDroppedLogger')
filtr.initialize()
records = [r for r in filtr.getRecords()]
filtr.finalize()
hashcode = hashlib.md5(open('tests/logger', 'rb').read()).hexdigest()
os.unlink('tests/logger')
self.assertEqual(hashcode, '3c30fe720f1b410fce6cf17651978b13')
