def test_filenames(self): for (filename, stem, ext) in split_filenames(): print(filename) expectation_filename = '%s.result' % stem with open(expectation_filename) as infile: expectation = eval(infile.read()) src = sources.Source(filename) self.assertEqual(list(src), expectation, msg="%s, by filename" % filename) # check that we can limit result size src = sources.Source(filename, limit=1) self.assertEqual(list(src), expectation[:1], msg='%s, limiting to 1') # now test against an open file object with sources._open(filename) as infile: src = sources.Source(infile) self.assertEqual(list(src), expectation, msg="%s, by file obj" % filename) # now test against the text contents if not filename.endswith('.pickle'): with open(filename) as infile: src = sources.Source(infile.read()) self.assertEqual(list(src), expectation, msg="%s, by contents" % filename)
""" Creates a .result file for each file in this directory, for verifying results. """ from file_stems import split_filenames from data_dispenser.sources import Source import pickle for (filename, stem, ext) in split_filenames(): src = Source(filename) src._dump('%s.result' % stem) src = Source(filename) with open('%s.pickle' % stem, 'wb') as outfile: pickle.dump(list(src), outfile)
""" Creates a .result file for each file in this directory, for verifying results. """ from file_stems import split_filenames from data_dispenser.sources import Source import pickle for (filename, stem, ext) in split_filenames(): src = Source(filename) src._dump('%s.result' % stem) src = Source(filename) if ext != '.xls' and ('.header' not in filename): with open('%s.pickle' % stem, 'wb') as outfile: pickle.dump(list(src), outfile)