def test_rewind(output_dir):
'test rewind calls base function and clear_output_dir'
output_dir = os.path.join(output_dir, 'clear_test')
r = Renderer(bna_sample, output_dir)
bg_name = r.background_map_name
fg_format = r.foreground_filename_format
# dump some files into output dir:
open(os.path.join(output_dir, bg_name), 'w').write('some junk')
for i in range(5):
open(os.path.join(output_dir,
fg_format.format(i)),
'w'
).write('some junk')
now = datetime.now()
r.prepare_for_model_run(model_start_time=now)
assert r._model_start_time == now
# prepare for model run clears output dir, but adds in the background map
files = sorted(os.listdir(output_dir))
assert files == sorted([os.path.basename(r.anim_filename),
r.background_map_name])
r.rewind()
# check super is called correctly
assert r._model_start_time is None
assert r._dt_since_lastoutput is None
assert r._write_step is True
# changed renderer and netcdf ouputter to delete old files in
# prepare_for_model_run() rather than rewind()
# -- rewind() was getting called a lot
# -- before there was time to change the ouput file names, etc.
# So for this unit test, there should only be a background image now.
files = sorted(os.listdir(output_dir))
assert files == sorted([os.path.basename(r.anim_filename),
r.background_map_name])
def test_rewind(output_dir):
'test rewind calls base function and clear_output_dir'
output_dir = os.path.join(output_dir, 'clear_test')
r = Renderer(bna_sample, output_dir)
bg_name = r.background_map_name
fg_format = r.foreground_filename_format
# dump some files into output dir:
open(os.path.join(output_dir, bg_name), 'w').write('some junk')
for i in range(5):
open(os.path.join(output_dir,
fg_format.format(i)),
'w'
).write('some junk')
now = datetime.now()
r.prepare_for_model_run(model_start_time=now)
assert r._model_start_time == now
# prepare for model run clears output dir, but adds in the background map
files = sorted(os.listdir(output_dir))
assert files == sorted([os.path.basename(r.anim_filename),
r.background_map_name])
r.rewind()
# check super is called correctly
assert r._model_start_time is None
assert r._dt_since_lastoutput is None
assert r._write_step is True
# changed renderer and netcdf ouputter to delete old files in
# prepare_for_model_run() rather than rewind()
# -- rewind() was getting called a lot
# -- before there was time to change the ouput file names, etc.
# So for this unit test, there should only be a background image now.
files = sorted(os.listdir(output_dir))
assert files == sorted([os.path.basename(r.anim_filename),
r.background_map_name])
def test_file_delete(output_dir):
output_dir = os.path.join(output_dir, 'clear_test')
r = Renderer(bna_sample, output_dir)
bg_name = r.background_map_name + "png"
fg_format = r.foreground_filename_format
# dump some files into output dir:
open(os.path.join(output_dir, bg_name), 'w').write('some junk')
for i in range(5):
open(os.path.join(output_dir,
fg_format.format(i) + "png"), 'w').write('some junk')
r.prepare_for_model_run(model_start_time=datetime.now())
# The default output formats are ['png','gif']
# so now there should only be a background image and the animated gif.
files = sorted(os.listdir(output_dir))
assert files == sorted(
[os.path.basename(r.anim_filename), r.background_map_name + "png"])
def test_file_delete(output_dir):
output_dir = os.path.join(output_dir, 'clear_test')
r = Renderer(bna_sample, output_dir)
bg_name = r.background_map_name
fg_format = r.foreground_filename_format
# dump some files into output dir:
open(os.path.join(output_dir, bg_name), 'w').write('some junk')
for i in range(5):
open(os.path.join(output_dir, fg_format.format(i)), 'w'
).write('some junk')
r.prepare_for_model_run(model_start_time=datetime.now())
# The default output formats are ['png','gif']
# so now there should only be a background image and the animated gif.
files = sorted(os.listdir(output_dir))
assert files == sorted([os.path.basename(r.anim_filename),
r.background_map_name])
