Skip to content
Snippets Groups Projects

Many changes

Merged Amir MOHAMMADI requested to merge amir into master
2 files
+ 41
27
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -19,8 +19,9 @@ def test_db_to_tfrecords():
try:
runner = CliRunner()
result = runner.invoke(db_to_tfrecords, args=(
dummy_config, '--output', output_path))
assert result.exit_code == 0, result.output
dummy_config, '--output', output_path), standalone_mode=False)
assert result.exit_code == 0, '%s\n%s\n%s' % (
result.exc_info, result.output, result.exception)
# TODO: test if the generated tfrecords file is equal with a reference
# file
@@ -29,18 +30,18 @@ def test_db_to_tfrecords():
shutil.rmtree(test_dir)
# def test_db_to_tfrecords_size_estimate():
# test_dir = tempfile.mkdtemp(prefix='bobtest_')
# output_path = os.path.join(test_dir, 'dev.tfrecords')
#
# try:
# runner = CliRunner()
# args = (dummy_config, '--size-estimate', '--output', output_path)
# print(' '.join(args))
# result = runner.invoke(db_to_tfrecords, args=args,)
# assert result.exit_code == 0, '%s\n%s\n%s' % (
# result.exc_info, result.output, result.exception)
# assert '2.0 M bytes' in result.output, result.output
#
# finally:
# shutil.rmtree(test_dir)
def test_db_to_tfrecords_size_estimate():
test_dir = tempfile.mkdtemp(prefix='bobtest_')
output_path = os.path.join(test_dir, 'dev.tfrecords')
try:
args = (dummy_config, '--size-estimate', '--output', output_path)
runner = CliRunner()
result = runner.invoke(
db_to_tfrecords, args=args, standalone_mode=False)
assert result.exit_code == 0, '%s\n%s\n%s' % (
result.exc_info, result.output, result.exception)
assert '2.0 M bytes' in result.output, result.output
finally:
shutil.rmtree(test_dir)
Loading