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)
