def set_config_test():
with scoped_curdir() as d:
test_cfg = d + '/test.cfg'
os.environ['STBT_CONFIG_FILE'] = test_cfg
with open(test_cfg, 'w') as f:
f.write(test_config)
yield
def set_config_test():
with scoped_curdir() as d:
test_cfg = d + '/test.cfg'
os.environ['STBT_CONFIG_FILE'] = test_cfg
with open(test_cfg, 'w') as f:
f.write(test_config)
yield
def test_ocr_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
f = stbt.load_image("action-panel.png")
r = stbt.Region(0, 370, right=1280, bottom=410)
c = (235, 235, 235)
with scoped_curdir(), scoped_debug_level(2):
stbt.ocr(f)
stbt.ocr(f, region=r)
stbt.ocr(f, region=r, text_color=c)
stbt.match_text("Summary", f) # no match
stbt.match_text("Summary", f, region=r) # no match
stbt.match_text("Summary", f, region=r, text_color=c)
files = subprocess.check_output("find stbt-debug | sort", shell=True) \
.decode("utf-8")
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/index.html
stbt-debug/00001/source.png
stbt-debug/00001/tessinput.png
stbt-debug/00001/upsampled.png
stbt-debug/00002
stbt-debug/00002/index.html
stbt-debug/00002/source.png
stbt-debug/00002/tessinput.png
stbt-debug/00002/upsampled.png
stbt-debug/00003
stbt-debug/00003/index.html
stbt-debug/00003/source.png
stbt-debug/00003/tessinput.png
stbt-debug/00003/text_color_difference.png
stbt-debug/00003/text_color_threshold.png
stbt-debug/00003/upsampled.png
stbt-debug/00004
stbt-debug/00004/index.html
stbt-debug/00004/source.png
stbt-debug/00004/tessinput.png
stbt-debug/00004/upsampled.png
stbt-debug/00005
stbt-debug/00005/index.html
stbt-debug/00005/source.png
stbt-debug/00005/tessinput.png
stbt-debug/00005/upsampled.png
stbt-debug/00006
stbt-debug/00006/index.html
stbt-debug/00006/source.png
stbt-debug/00006/tessinput.png
stbt-debug/00006/text_color_difference.png
stbt-debug/00006/text_color_threshold.png
stbt-debug/00006/upsampled.png
""")
def test_that_set_config_creates_directories_if_required():
with scoped_curdir() as d:
os.environ['XDG_CONFIG_HOME'] = d + '/.config'
if 'STBT_CONFIG_FILE' in os.environ:
del os.environ['STBT_CONFIG_FILE']
set_config('global', 'test', 'hello2')
assert os.path.isfile(d + '/.config/stbt/stbt.conf')
_config_init(force=True)
assert get_config('global', 'test') == 'hello2'
def test_that_set_config_creates_directories_if_required():
with scoped_curdir() as d:
os.environ['XDG_CONFIG_HOME'] = d + '/.config'
if 'STBT_CONFIG_FILE' in os.environ:
del os.environ['STBT_CONFIG_FILE']
set_config('global', 'test', 'hello2')
assert os.path.isfile(d + '/.config/stbt/stbt.conf')
_config_init(force=True)
assert get_config('global', 'test') == 'hello2'
def test_that_set_config_writes_to_the_first_stbt_config_file():
with scoped_curdir() as d:
filled_cfg = d + '/test.cfg'
empty_cfg = d + '/empty.cfg'
os.environ['STBT_CONFIG_FILE'] = '%s:%s' % (filled_cfg, empty_cfg)
open(filled_cfg, 'w')
open(empty_cfg, 'w')
set_config('global', 'test', 'goodbye')
assert open(filled_cfg).read().startswith('[global]')
assert open(empty_cfg).read() == ''
def test_that_set_config_writes_to_the_first_stbt_config_file():
with scoped_curdir() as d:
filled_cfg = d + '/test.cfg'
empty_cfg = d + '/empty.cfg'
os.environ['STBT_CONFIG_FILE'] = '%s:%s' % (filled_cfg, empty_cfg)
open(filled_cfg, 'w')
open(empty_cfg, 'w')
set_config('global', 'test', 'goodbye')
assert open(filled_cfg).read().startswith('[global]')
assert open(empty_cfg).read() == ''
def test_sponge_that_on_exception_file_isnt_modified():
with scoped_curdir():
open('foo', 'w').write('bar')
try:
with _sponge('foo') as f:
f.write('hello')
raise RuntimeError()
except RuntimeError:
pass
assert open('foo').read() == 'bar'
def test_sponge_that_on_exception_file_isnt_modified():
with scoped_curdir():
open('foo', 'w').write('bar')
try:
with _sponge('foo') as f:
f.write('hello')
raise RuntimeError()
except RuntimeError:
pass
assert open('foo').read() == 'bar'
def test_that_cache_is_disabled_when_debug_match():
# debug logging is a side effect that the cache cannot reproduce
import stbt
import _stbt.logging
with scoped_curdir() as srcdir, cache('cache.lmdb'):
stbt.match(srcdir + '/tests/red-black.png',
frame=numpy.zeros((720, 1280, 3), dtype=numpy.uint8))
assert not os.path.exists('stbt-debug')
with _stbt.logging.scoped_debug_level(2):
stbt.match(srcdir + '/tests/red-black.png',
frame=numpy.zeros((720, 1280, 3), dtype=numpy.uint8))
assert os.path.exists('stbt-debug')
def test_that_cache_is_disabled_when_debug_match():
# debug logging is a side effect that the cache cannot reproduce
import stbt
import _stbt.logging
with scoped_curdir() as srcdir, cache('cache.lmdb'):
stbt.match(srcdir + '/tests/red-black.png',
frame=numpy.zeros((720, 1280, 3), dtype=numpy.uint8))
assert not os.path.exists('stbt-debug')
with _stbt.logging.scoped_debug_level(2):
stbt.match(srcdir + '/tests/red-black.png',
frame=numpy.zeros((720, 1280, 3), dtype=numpy.uint8))
assert os.path.exists('stbt-debug')
def test_match_all_image_debug_when_first_pass_gives_no_matches():
with scoped_curdir(), scoped_debug_level(2):
matches = list(stbt.match_all(
"videotestsrc-redblue-flipped.png",
frame=_imread("videotestsrc-full-frame.png")))
print matches
assert len(matches) == 0
_verify_stbt_debug_dir([
'index.html',
'level2-source.png',
'level2-source_matchtemplate.png',
'level2-source_with_match.png',
'level2-source_with_rois.png',
'level2-template.png',
'match0-heatmap.png',
'match0-source_with_match.png',
'source.png',
'template.png',
])
def test_is_screen_black_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
f = stbt.load_image("videotestsrc-full-frame.png")
with scoped_curdir(), scoped_debug_level(2):
stbt.is_screen_black(f)
stbt.is_screen_black(f, mask="videotestsrc-mask-non-black.png")
stbt.is_screen_black(f, mask="videotestsrc-mask-no-video.png")
stbt.is_screen_black(f, region=stbt.Region(0, 0, 160, 120))
files = subprocess.check_output("find stbt-debug | sort", shell=True) \
.decode("utf-8")
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/grey.png
stbt-debug/00001/index.html
stbt-debug/00001/non_black.png
stbt-debug/00001/source.png
stbt-debug/00002
stbt-debug/00002/grey.png
stbt-debug/00002/index.html
stbt-debug/00002/mask.png
stbt-debug/00002/non_black.png
stbt-debug/00002/source.png
stbt-debug/00003
stbt-debug/00003/grey.png
stbt-debug/00003/index.html
stbt-debug/00003/mask.png
stbt-debug/00003/non_black.png
stbt-debug/00003/source.png
stbt-debug/00004
stbt-debug/00004/grey.png
stbt-debug/00004/index.html
stbt-debug/00004/non_black.png
stbt-debug/00004/source.png
""")
assert_expected("stbt-debug-expected-output/is_screen_black")
def test_is_screen_black_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
f = stbt.load_image("videotestsrc-full-frame.png")
with scoped_curdir(), scoped_debug_level(2):
stbt.is_screen_black(f)
stbt.is_screen_black(f, mask="videotestsrc-mask-non-black.png")
stbt.is_screen_black(f, mask="videotestsrc-mask-no-video.png")
stbt.is_screen_black(f, region=stbt.Region(0, 0, 160, 120))
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/grey.png
stbt-debug/00001/index.html
stbt-debug/00001/non_black.png
stbt-debug/00001/source.png
stbt-debug/00002
stbt-debug/00002/grey.png
stbt-debug/00002/index.html
stbt-debug/00002/mask.png
stbt-debug/00002/non_black.png
stbt-debug/00002/source.png
stbt-debug/00003
stbt-debug/00003/grey.png
stbt-debug/00003/index.html
stbt-debug/00003/mask.png
stbt-debug/00003/non_black.png
stbt-debug/00003/source.png
stbt-debug/00004
stbt-debug/00004/grey.png
stbt-debug/00004/index.html
stbt-debug/00004/non_black.png
stbt-debug/00004/source.png
""")
assert_expected("stbt-debug-expected-output/is_screen_black")
def test_match_all_image_debug_with_normed_absdiff():
with scoped_curdir(), scoped_debug_level(2):
matches = list(stbt.match_all(
"button.png", frame=_imread("buttons.png"),
match_parameters=mp(confirm_method="normed-absdiff",
confirm_threshold=0.3)))
print matches
assert len(matches) == 6
_verify_stbt_debug_dir([
'index.html',
'level0-source.png',
'level0-source_matchtemplate.png',
'level0-source_matchtemplate_threshold.png',
'level0-source_with_match.png',
'level0-source_with_rois.png',
'level0-template.png',
'level1-source.png',
'level1-source_matchtemplate.png',
'level1-source_matchtemplate_threshold.png',
'level1-source_with_match.png',
'level1-source_with_rois.png',
'level1-template.png',
'level2-source.png',
'level2-source_matchtemplate.png',
'level2-source_matchtemplate_threshold.png',
'level2-source_with_match.png',
'level2-source_with_rois.png',
'level2-template.png',
'match0-confirm-absdiff.png',
'match0-confirm-absdiff_threshold.png',
'match0-confirm-absdiff_threshold_erode.png',
'match0-confirm-source_roi.png',
'match0-confirm-source_roi_gray.png',
'match0-confirm-source_roi_gray_normalized.png',
'match0-confirm-template_gray.png',
'match0-confirm-template_gray_normalized.png',
'match0-heatmap.png',
'match0-source_with_match.png',
'match1-confirm-absdiff.png',
'match1-confirm-absdiff_threshold.png',
'match1-confirm-absdiff_threshold_erode.png',
'match1-confirm-source_roi.png',
'match1-confirm-source_roi_gray.png',
'match1-confirm-source_roi_gray_normalized.png',
'match1-confirm-template_gray.png',
'match1-confirm-template_gray_normalized.png',
'match1-heatmap.png',
'match1-source_with_match.png',
'match2-confirm-absdiff.png',
'match2-confirm-absdiff_threshold.png',
'match2-confirm-absdiff_threshold_erode.png',
'match2-confirm-source_roi.png',
'match2-confirm-source_roi_gray.png',
'match2-confirm-source_roi_gray_normalized.png',
'match2-confirm-template_gray.png',
'match2-confirm-template_gray_normalized.png',
'match2-heatmap.png',
'match2-source_with_match.png',
'match3-confirm-absdiff.png',
'match3-confirm-absdiff_threshold.png',
'match3-confirm-absdiff_threshold_erode.png',
'match3-confirm-source_roi.png',
'match3-confirm-source_roi_gray.png',
'match3-confirm-source_roi_gray_normalized.png',
'match3-confirm-template_gray.png',
'match3-confirm-template_gray_normalized.png',
'match3-heatmap.png',
'match3-source_with_match.png',
'match4-confirm-absdiff.png',
'match4-confirm-absdiff_threshold.png',
'match4-confirm-absdiff_threshold_erode.png',
'match4-confirm-source_roi.png',
'match4-confirm-source_roi_gray.png',
'match4-confirm-source_roi_gray_normalized.png',
'match4-confirm-template_gray.png',
'match4-confirm-template_gray_normalized.png',
'match4-heatmap.png',
'match4-source_with_match.png',
'match5-confirm-absdiff.png',
'match5-confirm-absdiff_threshold.png',
'match5-confirm-absdiff_threshold_erode.png',
'match5-confirm-source_roi.png',
'match5-confirm-source_roi_gray.png',
'match5-confirm-source_roi_gray_normalized.png',
'match5-confirm-template_gray.png',
'match5-confirm-template_gray_normalized.png',
'match5-heatmap.png',
'match5-source_with_match.png',
'match6-confirm-absdiff.png',
'match6-confirm-absdiff_threshold.png',
'match6-confirm-absdiff_threshold_erode.png',
'match6-confirm-source_roi.png',
'match6-confirm-source_roi_gray.png',
'match6-confirm-source_roi_gray_normalized.png',
'match6-confirm-template_gray.png',
'match6-confirm-template_gray_normalized.png',
'match6-heatmap.png',
'match6-source_with_match.png',
'source.png',
'template.png'
])
def test_match_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
with scoped_curdir(), scoped_debug_level(2):
# First pass gives no matches:
matches = list(stbt.match_all(
"videotestsrc-redblue-flipped.png",
frame=stbt.load_image("videotestsrc-full-frame.png")))
print matches
assert len(matches) == 0
# Multiple matches; first pass stops with a non-match:
matches = list(stbt.match_all(
"button.png", frame=stbt.load_image("buttons.png"),
match_parameters=mp(match_threshold=0.995)))
print matches
assert len(matches) == 6
# Multiple matches; second pass stops with a non-match:
matches = list(stbt.match_all(
"button.png", frame=stbt.load_image("buttons.png")))
print matches
assert len(matches) == 6
# With absdiff:
matches = list(stbt.match_all(
"button.png", frame=stbt.load_image("buttons.png"),
match_parameters=mp(confirm_method="absdiff",
confirm_threshold=0.84)))
print matches
assert len(matches) == 6
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/index.html
stbt-debug/00001/level2-source_matchtemplate.png
stbt-debug/00001/level2-source.png
stbt-debug/00001/level2-source_with_match.png
stbt-debug/00001/level2-source_with_rois.png
stbt-debug/00001/level2-template.png
stbt-debug/00001/match0-heatmap.png
stbt-debug/00001/match0-source_with_match.png
stbt-debug/00001/source.png
stbt-debug/00001/source_with_matches.png
stbt-debug/00001/template.png
stbt-debug/00002
stbt-debug/00002/index.html
stbt-debug/00002/level0-source_matchtemplate.png
stbt-debug/00002/level0-source_matchtemplate_threshold.png
stbt-debug/00002/level0-source.png
stbt-debug/00002/level0-source_with_match.png
stbt-debug/00002/level0-source_with_rois.png
stbt-debug/00002/level0-template.png
stbt-debug/00002/level1-source_matchtemplate.png
stbt-debug/00002/level1-source_matchtemplate_threshold.png
stbt-debug/00002/level1-source.png
stbt-debug/00002/level1-source_with_match.png
stbt-debug/00002/level1-source_with_rois.png
stbt-debug/00002/level1-template.png
stbt-debug/00002/level2-source_matchtemplate.png
stbt-debug/00002/level2-source_matchtemplate_threshold.png
stbt-debug/00002/level2-source.png
stbt-debug/00002/level2-source_with_match.png
stbt-debug/00002/level2-source_with_rois.png
stbt-debug/00002/level2-template.png
stbt-debug/00002/match0-confirm-absdiff.png
stbt-debug/00002/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match0-confirm-absdiff_threshold.png
stbt-debug/00002/match0-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match0-confirm-source_roi_gray.png
stbt-debug/00002/match0-confirm-source_roi.png
stbt-debug/00002/match0-confirm-template_gray_normalized.png
stbt-debug/00002/match0-confirm-template_gray.png
stbt-debug/00002/match0-heatmap.png
stbt-debug/00002/match0-source_with_match.png
stbt-debug/00002/match1-confirm-absdiff.png
stbt-debug/00002/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match1-confirm-absdiff_threshold.png
stbt-debug/00002/match1-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match1-confirm-source_roi_gray.png
stbt-debug/00002/match1-confirm-source_roi.png
stbt-debug/00002/match1-confirm-template_gray_normalized.png
stbt-debug/00002/match1-confirm-template_gray.png
stbt-debug/00002/match1-heatmap.png
stbt-debug/00002/match1-source_with_match.png
stbt-debug/00002/match2-confirm-absdiff.png
stbt-debug/00002/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match2-confirm-absdiff_threshold.png
stbt-debug/00002/match2-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match2-confirm-source_roi_gray.png
stbt-debug/00002/match2-confirm-source_roi.png
stbt-debug/00002/match2-confirm-template_gray_normalized.png
stbt-debug/00002/match2-confirm-template_gray.png
stbt-debug/00002/match2-heatmap.png
stbt-debug/00002/match2-source_with_match.png
stbt-debug/00002/match3-confirm-absdiff.png
stbt-debug/00002/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match3-confirm-absdiff_threshold.png
stbt-debug/00002/match3-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match3-confirm-source_roi_gray.png
stbt-debug/00002/match3-confirm-source_roi.png
stbt-debug/00002/match3-confirm-template_gray_normalized.png
stbt-debug/00002/match3-confirm-template_gray.png
stbt-debug/00002/match3-heatmap.png
stbt-debug/00002/match3-source_with_match.png
stbt-debug/00002/match4-confirm-absdiff.png
stbt-debug/00002/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match4-confirm-absdiff_threshold.png
stbt-debug/00002/match4-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match4-confirm-source_roi_gray.png
stbt-debug/00002/match4-confirm-source_roi.png
stbt-debug/00002/match4-confirm-template_gray_normalized.png
stbt-debug/00002/match4-confirm-template_gray.png
stbt-debug/00002/match4-heatmap.png
stbt-debug/00002/match4-source_with_match.png
stbt-debug/00002/match5-confirm-absdiff.png
stbt-debug/00002/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match5-confirm-absdiff_threshold.png
stbt-debug/00002/match5-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match5-confirm-source_roi_gray.png
stbt-debug/00002/match5-confirm-source_roi.png
stbt-debug/00002/match5-confirm-template_gray_normalized.png
stbt-debug/00002/match5-confirm-template_gray.png
stbt-debug/00002/match5-heatmap.png
stbt-debug/00002/match5-source_with_match.png
stbt-debug/00002/match6-heatmap.png
stbt-debug/00002/match6-source_with_match.png
stbt-debug/00002/source.png
stbt-debug/00002/source_with_matches.png
stbt-debug/00002/template.png
stbt-debug/00003
stbt-debug/00003/index.html
stbt-debug/00003/level0-source_matchtemplate.png
stbt-debug/00003/level0-source_matchtemplate_threshold.png
stbt-debug/00003/level0-source.png
stbt-debug/00003/level0-source_with_match.png
stbt-debug/00003/level0-source_with_rois.png
stbt-debug/00003/level0-template.png
stbt-debug/00003/level1-source_matchtemplate.png
stbt-debug/00003/level1-source_matchtemplate_threshold.png
stbt-debug/00003/level1-source.png
stbt-debug/00003/level1-source_with_match.png
stbt-debug/00003/level1-source_with_rois.png
stbt-debug/00003/level1-template.png
stbt-debug/00003/level2-source_matchtemplate.png
stbt-debug/00003/level2-source_matchtemplate_threshold.png
stbt-debug/00003/level2-source.png
stbt-debug/00003/level2-source_with_match.png
stbt-debug/00003/level2-source_with_rois.png
stbt-debug/00003/level2-template.png
stbt-debug/00003/match0-confirm-absdiff.png
stbt-debug/00003/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match0-confirm-absdiff_threshold.png
stbt-debug/00003/match0-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match0-confirm-source_roi_gray.png
stbt-debug/00003/match0-confirm-source_roi.png
stbt-debug/00003/match0-confirm-template_gray_normalized.png
stbt-debug/00003/match0-confirm-template_gray.png
stbt-debug/00003/match0-heatmap.png
stbt-debug/00003/match0-source_with_match.png
stbt-debug/00003/match1-confirm-absdiff.png
stbt-debug/00003/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match1-confirm-absdiff_threshold.png
stbt-debug/00003/match1-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match1-confirm-source_roi_gray.png
stbt-debug/00003/match1-confirm-source_roi.png
stbt-debug/00003/match1-confirm-template_gray_normalized.png
stbt-debug/00003/match1-confirm-template_gray.png
stbt-debug/00003/match1-heatmap.png
stbt-debug/00003/match1-source_with_match.png
stbt-debug/00003/match2-confirm-absdiff.png
stbt-debug/00003/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match2-confirm-absdiff_threshold.png
stbt-debug/00003/match2-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match2-confirm-source_roi_gray.png
stbt-debug/00003/match2-confirm-source_roi.png
stbt-debug/00003/match2-confirm-template_gray_normalized.png
stbt-debug/00003/match2-confirm-template_gray.png
stbt-debug/00003/match2-heatmap.png
stbt-debug/00003/match2-source_with_match.png
stbt-debug/00003/match3-confirm-absdiff.png
stbt-debug/00003/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match3-confirm-absdiff_threshold.png
stbt-debug/00003/match3-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match3-confirm-source_roi_gray.png
stbt-debug/00003/match3-confirm-source_roi.png
stbt-debug/00003/match3-confirm-template_gray_normalized.png
stbt-debug/00003/match3-confirm-template_gray.png
stbt-debug/00003/match3-heatmap.png
stbt-debug/00003/match3-source_with_match.png
stbt-debug/00003/match4-confirm-absdiff.png
stbt-debug/00003/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match4-confirm-absdiff_threshold.png
stbt-debug/00003/match4-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match4-confirm-source_roi_gray.png
stbt-debug/00003/match4-confirm-source_roi.png
stbt-debug/00003/match4-confirm-template_gray_normalized.png
stbt-debug/00003/match4-confirm-template_gray.png
stbt-debug/00003/match4-heatmap.png
stbt-debug/00003/match4-source_with_match.png
stbt-debug/00003/match5-confirm-absdiff.png
stbt-debug/00003/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match5-confirm-absdiff_threshold.png
stbt-debug/00003/match5-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match5-confirm-source_roi_gray.png
stbt-debug/00003/match5-confirm-source_roi.png
stbt-debug/00003/match5-confirm-template_gray_normalized.png
stbt-debug/00003/match5-confirm-template_gray.png
stbt-debug/00003/match5-heatmap.png
stbt-debug/00003/match5-source_with_match.png
stbt-debug/00003/match6-confirm-absdiff.png
stbt-debug/00003/match6-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match6-confirm-absdiff_threshold.png
stbt-debug/00003/match6-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match6-confirm-source_roi_gray.png
stbt-debug/00003/match6-confirm-source_roi.png
stbt-debug/00003/match6-confirm-template_gray_normalized.png
stbt-debug/00003/match6-confirm-template_gray.png
stbt-debug/00003/match6-heatmap.png
stbt-debug/00003/match6-source_with_match.png
stbt-debug/00003/source.png
stbt-debug/00003/source_with_matches.png
stbt-debug/00003/template.png
stbt-debug/00004
stbt-debug/00004/index.html
stbt-debug/00004/level0-source_matchtemplate.png
stbt-debug/00004/level0-source_matchtemplate_threshold.png
stbt-debug/00004/level0-source.png
stbt-debug/00004/level0-source_with_match.png
stbt-debug/00004/level0-source_with_rois.png
stbt-debug/00004/level0-template.png
stbt-debug/00004/level1-source_matchtemplate.png
stbt-debug/00004/level1-source_matchtemplate_threshold.png
stbt-debug/00004/level1-source.png
stbt-debug/00004/level1-source_with_match.png
stbt-debug/00004/level1-source_with_rois.png
stbt-debug/00004/level1-template.png
stbt-debug/00004/level2-source_matchtemplate.png
stbt-debug/00004/level2-source_matchtemplate_threshold.png
stbt-debug/00004/level2-source.png
stbt-debug/00004/level2-source_with_match.png
stbt-debug/00004/level2-source_with_rois.png
stbt-debug/00004/level2-template.png
stbt-debug/00004/match0-confirm-absdiff.png
stbt-debug/00004/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match0-confirm-absdiff_threshold.png
stbt-debug/00004/match0-confirm-source_roi_gray.png
stbt-debug/00004/match0-confirm-source_roi.png
stbt-debug/00004/match0-confirm-template_gray.png
stbt-debug/00004/match0-heatmap.png
stbt-debug/00004/match0-source_with_match.png
stbt-debug/00004/match1-confirm-absdiff.png
stbt-debug/00004/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match1-confirm-absdiff_threshold.png
stbt-debug/00004/match1-confirm-source_roi_gray.png
stbt-debug/00004/match1-confirm-source_roi.png
stbt-debug/00004/match1-confirm-template_gray.png
stbt-debug/00004/match1-heatmap.png
stbt-debug/00004/match1-source_with_match.png
stbt-debug/00004/match2-confirm-absdiff.png
stbt-debug/00004/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match2-confirm-absdiff_threshold.png
stbt-debug/00004/match2-confirm-source_roi_gray.png
stbt-debug/00004/match2-confirm-source_roi.png
stbt-debug/00004/match2-confirm-template_gray.png
stbt-debug/00004/match2-heatmap.png
stbt-debug/00004/match2-source_with_match.png
stbt-debug/00004/match3-confirm-absdiff.png
stbt-debug/00004/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match3-confirm-absdiff_threshold.png
stbt-debug/00004/match3-confirm-source_roi_gray.png
stbt-debug/00004/match3-confirm-source_roi.png
stbt-debug/00004/match3-confirm-template_gray.png
stbt-debug/00004/match3-heatmap.png
stbt-debug/00004/match3-source_with_match.png
stbt-debug/00004/match4-confirm-absdiff.png
stbt-debug/00004/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match4-confirm-absdiff_threshold.png
stbt-debug/00004/match4-confirm-source_roi_gray.png
stbt-debug/00004/match4-confirm-source_roi.png
stbt-debug/00004/match4-confirm-template_gray.png
stbt-debug/00004/match4-heatmap.png
stbt-debug/00004/match4-source_with_match.png
stbt-debug/00004/match5-confirm-absdiff.png
stbt-debug/00004/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match5-confirm-absdiff_threshold.png
stbt-debug/00004/match5-confirm-source_roi_gray.png
stbt-debug/00004/match5-confirm-source_roi.png
stbt-debug/00004/match5-confirm-template_gray.png
stbt-debug/00004/match5-heatmap.png
stbt-debug/00004/match5-source_with_match.png
stbt-debug/00004/match6-confirm-absdiff.png
stbt-debug/00004/match6-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match6-confirm-absdiff_threshold.png
stbt-debug/00004/match6-confirm-source_roi_gray.png
stbt-debug/00004/match6-confirm-source_roi.png
stbt-debug/00004/match6-confirm-template_gray.png
stbt-debug/00004/match6-heatmap.png
stbt-debug/00004/match6-source_with_match.png
stbt-debug/00004/source.png
stbt-debug/00004/source_with_matches.png
stbt-debug/00004/template.png
""")
assert_expected("stbt-debug-expected-output/match")
def test_motion_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
def fake_frames():
for i, f in enumerate(["box-00001.png",
"box-00002.png",
"box-00003.png"]):
yield stbt.Frame(stbt.load_image(f), time=i)
with scoped_curdir(), scoped_debug_level(2):
for _ in stbt.detect_motion(frames=fake_frames()):
pass
for _ in stbt.detect_motion(frames=fake_frames(), mask="box-00000.png"):
pass
for _ in stbt.detect_motion(frames=fake_frames(),
region=stbt.Region(0, 0, 320, 400)):
pass
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/absdiff.png
stbt-debug/00001/absdiff_threshold_erode.png
stbt-debug/00001/absdiff_threshold.png
stbt-debug/00001/gray.png
stbt-debug/00001/index.html
stbt-debug/00001/previous_frame_gray.png
stbt-debug/00001/source.png
stbt-debug/00002
stbt-debug/00002/absdiff.png
stbt-debug/00002/absdiff_threshold_erode.png
stbt-debug/00002/absdiff_threshold.png
stbt-debug/00002/gray.png
stbt-debug/00002/index.html
stbt-debug/00002/previous_frame_gray.png
stbt-debug/00002/source.png
stbt-debug/00003
stbt-debug/00003/absdiff_masked.png
stbt-debug/00003/absdiff.png
stbt-debug/00003/absdiff_threshold_erode.png
stbt-debug/00003/absdiff_threshold.png
stbt-debug/00003/gray.png
stbt-debug/00003/index.html
stbt-debug/00003/mask.png
stbt-debug/00003/previous_frame_gray.png
stbt-debug/00003/source.png
stbt-debug/00004
stbt-debug/00004/absdiff_masked.png
stbt-debug/00004/absdiff.png
stbt-debug/00004/absdiff_threshold_erode.png
stbt-debug/00004/absdiff_threshold.png
stbt-debug/00004/gray.png
stbt-debug/00004/index.html
stbt-debug/00004/mask.png
stbt-debug/00004/previous_frame_gray.png
stbt-debug/00004/source.png
stbt-debug/00005
stbt-debug/00005/absdiff.png
stbt-debug/00005/absdiff_threshold_erode.png
stbt-debug/00005/absdiff_threshold.png
stbt-debug/00005/gray.png
stbt-debug/00005/index.html
stbt-debug/00005/previous_frame_gray.png
stbt-debug/00005/source.png
stbt-debug/00006
stbt-debug/00006/absdiff.png
stbt-debug/00006/absdiff_threshold_erode.png
stbt-debug/00006/absdiff_threshold.png
stbt-debug/00006/gray.png
stbt-debug/00006/index.html
stbt-debug/00006/previous_frame_gray.png
stbt-debug/00006/source.png
""")
assert_expected("stbt-debug-expected-output/motion")
def test_motion_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
def fake_frames():
for i, f in enumerate(
["box-00001.png", "box-00002.png", "box-00003.png"]):
yield stbt.Frame(stbt.load_image(f), time=i)
with scoped_curdir(), scoped_debug_level(2):
for _ in stbt.detect_motion(frames=fake_frames()):
pass
for _ in stbt.detect_motion(frames=fake_frames(),
mask="box-00000.png"):
pass
for _ in stbt.detect_motion(frames=fake_frames(),
region=stbt.Region(0, 0, 320, 400)):
pass
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/absdiff.png
stbt-debug/00001/absdiff_threshold_erode.png
stbt-debug/00001/absdiff_threshold.png
stbt-debug/00001/gray.png
stbt-debug/00001/index.html
stbt-debug/00001/previous_frame_gray.png
stbt-debug/00001/source.png
stbt-debug/00002
stbt-debug/00002/absdiff.png
stbt-debug/00002/absdiff_threshold_erode.png
stbt-debug/00002/absdiff_threshold.png
stbt-debug/00002/gray.png
stbt-debug/00002/index.html
stbt-debug/00002/previous_frame_gray.png
stbt-debug/00002/source.png
stbt-debug/00003
stbt-debug/00003/absdiff_masked.png
stbt-debug/00003/absdiff.png
stbt-debug/00003/absdiff_threshold_erode.png
stbt-debug/00003/absdiff_threshold.png
stbt-debug/00003/gray.png
stbt-debug/00003/index.html
stbt-debug/00003/mask.png
stbt-debug/00003/previous_frame_gray.png
stbt-debug/00003/source.png
stbt-debug/00004
stbt-debug/00004/absdiff_masked.png
stbt-debug/00004/absdiff.png
stbt-debug/00004/absdiff_threshold_erode.png
stbt-debug/00004/absdiff_threshold.png
stbt-debug/00004/gray.png
stbt-debug/00004/index.html
stbt-debug/00004/mask.png
stbt-debug/00004/previous_frame_gray.png
stbt-debug/00004/source.png
stbt-debug/00005
stbt-debug/00005/absdiff.png
stbt-debug/00005/absdiff_threshold_erode.png
stbt-debug/00005/absdiff_threshold.png
stbt-debug/00005/gray.png
stbt-debug/00005/index.html
stbt-debug/00005/previous_frame_gray.png
stbt-debug/00005/source.png
stbt-debug/00006
stbt-debug/00006/absdiff.png
stbt-debug/00006/absdiff_threshold_erode.png
stbt-debug/00006/absdiff_threshold.png
stbt-debug/00006/gray.png
stbt-debug/00006/index.html
stbt-debug/00006/previous_frame_gray.png
stbt-debug/00006/source.png
""")
assert_expected("stbt-debug-expected-output/motion")
def test_sponge_that_new_data_end_up_in_file():
with scoped_curdir():
with _sponge('hello') as f:
f.write('hello')
assert open('hello').read() == 'hello'
def test_match_all_image_debug_when_second_pass_stops_with_a_nonmatch():
with scoped_curdir(), scoped_debug_level(2):
matches = list(stbt.match_all(
"button.png", frame=_imread("buttons.png")))
print matches
assert len(matches) == 6
_verify_stbt_debug_dir([
'index.html',
'level0-source.png',
'level0-source_matchtemplate.png',
'level0-source_matchtemplate_threshold.png',
'level0-source_with_match.png',
'level0-source_with_rois.png',
'level0-template.png',
'level1-source.png',
'level1-source_matchtemplate.png',
'level1-source_matchtemplate_threshold.png',
'level1-source_with_match.png',
'level1-source_with_rois.png',
'level1-template.png',
'level2-source.png',
'level2-source_matchtemplate.png',
'level2-source_matchtemplate_threshold.png',
'level2-source_with_match.png',
'level2-source_with_rois.png',
'level2-template.png',
'match0-confirm-absdiff.png',
'match0-confirm-absdiff_threshold.png',
'match0-confirm-absdiff_threshold_erode.png',
'match0-confirm-source_roi.png',
'match0-confirm-source_roi_gray.png',
'match0-confirm-template_gray.png',
'match0-heatmap.png',
'match0-source_with_match.png',
'match1-confirm-absdiff.png',
'match1-confirm-absdiff_threshold.png',
'match1-confirm-absdiff_threshold_erode.png',
'match1-confirm-source_roi.png',
'match1-confirm-source_roi_gray.png',
'match1-confirm-template_gray.png',
'match1-heatmap.png',
'match1-source_with_match.png',
'match2-confirm-absdiff.png',
'match2-confirm-absdiff_threshold.png',
'match2-confirm-absdiff_threshold_erode.png',
'match2-confirm-source_roi.png',
'match2-confirm-source_roi_gray.png',
'match2-confirm-template_gray.png',
'match2-heatmap.png',
'match2-source_with_match.png',
'match3-confirm-absdiff.png',
'match3-confirm-absdiff_threshold.png',
'match3-confirm-absdiff_threshold_erode.png',
'match3-confirm-source_roi.png',
'match3-confirm-source_roi_gray.png',
'match3-confirm-template_gray.png',
'match3-heatmap.png',
'match3-source_with_match.png',
'match4-confirm-absdiff.png',
'match4-confirm-absdiff_threshold.png',
'match4-confirm-absdiff_threshold_erode.png',
'match4-confirm-source_roi.png',
'match4-confirm-source_roi_gray.png',
'match4-confirm-template_gray.png',
'match4-heatmap.png',
'match4-source_with_match.png',
'match5-confirm-absdiff.png',
'match5-confirm-absdiff_threshold.png',
'match5-confirm-absdiff_threshold_erode.png',
'match5-confirm-source_roi.png',
'match5-confirm-source_roi_gray.png',
'match5-confirm-template_gray.png',
'match5-heatmap.png',
'match5-source_with_match.png',
'match6-confirm-absdiff.png',
'match6-confirm-absdiff_threshold.png',
'match6-confirm-absdiff_threshold_erode.png',
'match6-confirm-source_roi.png',
'match6-confirm-source_roi_gray.png',
'match6-confirm-template_gray.png',
'match6-heatmap.png',
'match6-source_with_match.png',
'source.png',
'template.png',
])
def test_ocr_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
f = stbt.load_image("action-panel.png")
r = stbt.Region(0, 370, right=1280, bottom=410)
c = (235, 235, 235)
nonoverlapping = stbt.Region(2000, 2000, width=10, height=10)
with scoped_curdir(), scoped_debug_level(2):
stbt.ocr(f)
stbt.ocr(f, region=r)
stbt.ocr(f, region=r, text_color=c)
stbt.ocr(f, region=nonoverlapping)
stbt.match_text("Summary", f) # no match
stbt.match_text("Summary", f, region=r) # no match
stbt.match_text("Summary", f, region=r, text_color=c)
stbt.match_text("Summary", f, region=nonoverlapping)
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/index.html
stbt-debug/00001/source.png
stbt-debug/00001/tessinput.png
stbt-debug/00001/upsampled.png
stbt-debug/00002
stbt-debug/00002/index.html
stbt-debug/00002/source.png
stbt-debug/00002/tessinput.png
stbt-debug/00002/upsampled.png
stbt-debug/00003
stbt-debug/00003/index.html
stbt-debug/00003/source.png
stbt-debug/00003/tessinput.png
stbt-debug/00003/text_color_difference.png
stbt-debug/00003/text_color_threshold.png
stbt-debug/00003/upsampled.png
stbt-debug/00004
stbt-debug/00004/index.html
stbt-debug/00004/source.png
stbt-debug/00005
stbt-debug/00005/index.html
stbt-debug/00005/source.png
stbt-debug/00005/tessinput.png
stbt-debug/00005/upsampled.png
stbt-debug/00006
stbt-debug/00006/index.html
stbt-debug/00006/source.png
stbt-debug/00006/tessinput.png
stbt-debug/00006/upsampled.png
stbt-debug/00007
stbt-debug/00007/index.html
stbt-debug/00007/source.png
stbt-debug/00007/tessinput.png
stbt-debug/00007/text_color_difference.png
stbt-debug/00007/text_color_threshold.png
stbt-debug/00007/upsampled.png
stbt-debug/00008
stbt-debug/00008/index.html
stbt-debug/00008/source.png
""")
def test_match_debug():
# So that the output directory name doesn't depend on how many tests
# were run before this one.
ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access
with scoped_curdir(), scoped_debug_level(2):
# First pass gives no matches:
matches = list(
stbt.match_all(
"videotestsrc-redblue-flipped.png",
frame=stbt.load_image("videotestsrc-full-frame.png")))
print matches
assert len(matches) == 0
# Multiple matches; first pass stops with a non-match:
matches = list(
stbt.match_all("button.png",
frame=stbt.load_image("buttons.png"),
match_parameters=mp(match_threshold=0.995)))
print matches
assert len(matches) == 6
# Multiple matches; second pass stops with a non-match:
matches = list(
stbt.match_all("button.png", frame=stbt.load_image("buttons.png")))
print matches
assert len(matches) == 6
# With absdiff:
matches = list(
stbt.match_all("button.png",
frame=stbt.load_image("buttons.png"),
match_parameters=mp(confirm_method="absdiff",
confirm_threshold=0.16)))
print matches
assert len(matches) == 6
files = subprocess.check_output("find stbt-debug | sort", shell=True)
assert files == dedent("""\
stbt-debug
stbt-debug/00001
stbt-debug/00001/index.html
stbt-debug/00001/level2-source_matchtemplate.png
stbt-debug/00001/level2-source.png
stbt-debug/00001/level2-source_with_match.png
stbt-debug/00001/level2-source_with_rois.png
stbt-debug/00001/level2-template.png
stbt-debug/00001/match0-heatmap.png
stbt-debug/00001/match0-source_with_match.png
stbt-debug/00001/source.png
stbt-debug/00001/source_with_matches.png
stbt-debug/00001/template.png
stbt-debug/00002
stbt-debug/00002/index.html
stbt-debug/00002/level0-source_matchtemplate.png
stbt-debug/00002/level0-source_matchtemplate_threshold.png
stbt-debug/00002/level0-source.png
stbt-debug/00002/level0-source_with_match.png
stbt-debug/00002/level0-source_with_rois.png
stbt-debug/00002/level0-template.png
stbt-debug/00002/level1-source_matchtemplate.png
stbt-debug/00002/level1-source_matchtemplate_threshold.png
stbt-debug/00002/level1-source.png
stbt-debug/00002/level1-source_with_match.png
stbt-debug/00002/level1-source_with_rois.png
stbt-debug/00002/level1-template.png
stbt-debug/00002/level2-source_matchtemplate.png
stbt-debug/00002/level2-source_matchtemplate_threshold.png
stbt-debug/00002/level2-source.png
stbt-debug/00002/level2-source_with_match.png
stbt-debug/00002/level2-source_with_rois.png
stbt-debug/00002/level2-template.png
stbt-debug/00002/match0-confirm-absdiff.png
stbt-debug/00002/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match0-confirm-absdiff_threshold.png
stbt-debug/00002/match0-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match0-confirm-source_roi_gray.png
stbt-debug/00002/match0-confirm-source_roi.png
stbt-debug/00002/match0-confirm-template_gray_normalized.png
stbt-debug/00002/match0-confirm-template_gray.png
stbt-debug/00002/match0-heatmap.png
stbt-debug/00002/match0-source_with_match.png
stbt-debug/00002/match1-confirm-absdiff.png
stbt-debug/00002/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match1-confirm-absdiff_threshold.png
stbt-debug/00002/match1-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match1-confirm-source_roi_gray.png
stbt-debug/00002/match1-confirm-source_roi.png
stbt-debug/00002/match1-confirm-template_gray_normalized.png
stbt-debug/00002/match1-confirm-template_gray.png
stbt-debug/00002/match1-heatmap.png
stbt-debug/00002/match1-source_with_match.png
stbt-debug/00002/match2-confirm-absdiff.png
stbt-debug/00002/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match2-confirm-absdiff_threshold.png
stbt-debug/00002/match2-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match2-confirm-source_roi_gray.png
stbt-debug/00002/match2-confirm-source_roi.png
stbt-debug/00002/match2-confirm-template_gray_normalized.png
stbt-debug/00002/match2-confirm-template_gray.png
stbt-debug/00002/match2-heatmap.png
stbt-debug/00002/match2-source_with_match.png
stbt-debug/00002/match3-confirm-absdiff.png
stbt-debug/00002/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match3-confirm-absdiff_threshold.png
stbt-debug/00002/match3-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match3-confirm-source_roi_gray.png
stbt-debug/00002/match3-confirm-source_roi.png
stbt-debug/00002/match3-confirm-template_gray_normalized.png
stbt-debug/00002/match3-confirm-template_gray.png
stbt-debug/00002/match3-heatmap.png
stbt-debug/00002/match3-source_with_match.png
stbt-debug/00002/match4-confirm-absdiff.png
stbt-debug/00002/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match4-confirm-absdiff_threshold.png
stbt-debug/00002/match4-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match4-confirm-source_roi_gray.png
stbt-debug/00002/match4-confirm-source_roi.png
stbt-debug/00002/match4-confirm-template_gray_normalized.png
stbt-debug/00002/match4-confirm-template_gray.png
stbt-debug/00002/match4-heatmap.png
stbt-debug/00002/match4-source_with_match.png
stbt-debug/00002/match5-confirm-absdiff.png
stbt-debug/00002/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00002/match5-confirm-absdiff_threshold.png
stbt-debug/00002/match5-confirm-source_roi_gray_normalized.png
stbt-debug/00002/match5-confirm-source_roi_gray.png
stbt-debug/00002/match5-confirm-source_roi.png
stbt-debug/00002/match5-confirm-template_gray_normalized.png
stbt-debug/00002/match5-confirm-template_gray.png
stbt-debug/00002/match5-heatmap.png
stbt-debug/00002/match5-source_with_match.png
stbt-debug/00002/match6-heatmap.png
stbt-debug/00002/match6-source_with_match.png
stbt-debug/00002/source.png
stbt-debug/00002/source_with_matches.png
stbt-debug/00002/template.png
stbt-debug/00003
stbt-debug/00003/index.html
stbt-debug/00003/level0-source_matchtemplate.png
stbt-debug/00003/level0-source_matchtemplate_threshold.png
stbt-debug/00003/level0-source.png
stbt-debug/00003/level0-source_with_match.png
stbt-debug/00003/level0-source_with_rois.png
stbt-debug/00003/level0-template.png
stbt-debug/00003/level1-source_matchtemplate.png
stbt-debug/00003/level1-source_matchtemplate_threshold.png
stbt-debug/00003/level1-source.png
stbt-debug/00003/level1-source_with_match.png
stbt-debug/00003/level1-source_with_rois.png
stbt-debug/00003/level1-template.png
stbt-debug/00003/level2-source_matchtemplate.png
stbt-debug/00003/level2-source_matchtemplate_threshold.png
stbt-debug/00003/level2-source.png
stbt-debug/00003/level2-source_with_match.png
stbt-debug/00003/level2-source_with_rois.png
stbt-debug/00003/level2-template.png
stbt-debug/00003/match0-confirm-absdiff.png
stbt-debug/00003/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match0-confirm-absdiff_threshold.png
stbt-debug/00003/match0-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match0-confirm-source_roi_gray.png
stbt-debug/00003/match0-confirm-source_roi.png
stbt-debug/00003/match0-confirm-template_gray_normalized.png
stbt-debug/00003/match0-confirm-template_gray.png
stbt-debug/00003/match0-heatmap.png
stbt-debug/00003/match0-source_with_match.png
stbt-debug/00003/match1-confirm-absdiff.png
stbt-debug/00003/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match1-confirm-absdiff_threshold.png
stbt-debug/00003/match1-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match1-confirm-source_roi_gray.png
stbt-debug/00003/match1-confirm-source_roi.png
stbt-debug/00003/match1-confirm-template_gray_normalized.png
stbt-debug/00003/match1-confirm-template_gray.png
stbt-debug/00003/match1-heatmap.png
stbt-debug/00003/match1-source_with_match.png
stbt-debug/00003/match2-confirm-absdiff.png
stbt-debug/00003/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match2-confirm-absdiff_threshold.png
stbt-debug/00003/match2-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match2-confirm-source_roi_gray.png
stbt-debug/00003/match2-confirm-source_roi.png
stbt-debug/00003/match2-confirm-template_gray_normalized.png
stbt-debug/00003/match2-confirm-template_gray.png
stbt-debug/00003/match2-heatmap.png
stbt-debug/00003/match2-source_with_match.png
stbt-debug/00003/match3-confirm-absdiff.png
stbt-debug/00003/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match3-confirm-absdiff_threshold.png
stbt-debug/00003/match3-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match3-confirm-source_roi_gray.png
stbt-debug/00003/match3-confirm-source_roi.png
stbt-debug/00003/match3-confirm-template_gray_normalized.png
stbt-debug/00003/match3-confirm-template_gray.png
stbt-debug/00003/match3-heatmap.png
stbt-debug/00003/match3-source_with_match.png
stbt-debug/00003/match4-confirm-absdiff.png
stbt-debug/00003/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match4-confirm-absdiff_threshold.png
stbt-debug/00003/match4-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match4-confirm-source_roi_gray.png
stbt-debug/00003/match4-confirm-source_roi.png
stbt-debug/00003/match4-confirm-template_gray_normalized.png
stbt-debug/00003/match4-confirm-template_gray.png
stbt-debug/00003/match4-heatmap.png
stbt-debug/00003/match4-source_with_match.png
stbt-debug/00003/match5-confirm-absdiff.png
stbt-debug/00003/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match5-confirm-absdiff_threshold.png
stbt-debug/00003/match5-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match5-confirm-source_roi_gray.png
stbt-debug/00003/match5-confirm-source_roi.png
stbt-debug/00003/match5-confirm-template_gray_normalized.png
stbt-debug/00003/match5-confirm-template_gray.png
stbt-debug/00003/match5-heatmap.png
stbt-debug/00003/match5-source_with_match.png
stbt-debug/00003/match6-confirm-absdiff.png
stbt-debug/00003/match6-confirm-absdiff_threshold_erode.png
stbt-debug/00003/match6-confirm-absdiff_threshold.png
stbt-debug/00003/match6-confirm-source_roi_gray_normalized.png
stbt-debug/00003/match6-confirm-source_roi_gray.png
stbt-debug/00003/match6-confirm-source_roi.png
stbt-debug/00003/match6-confirm-template_gray_normalized.png
stbt-debug/00003/match6-confirm-template_gray.png
stbt-debug/00003/match6-heatmap.png
stbt-debug/00003/match6-source_with_match.png
stbt-debug/00003/source.png
stbt-debug/00003/source_with_matches.png
stbt-debug/00003/template.png
stbt-debug/00004
stbt-debug/00004/index.html
stbt-debug/00004/level0-source_matchtemplate.png
stbt-debug/00004/level0-source_matchtemplate_threshold.png
stbt-debug/00004/level0-source.png
stbt-debug/00004/level0-source_with_match.png
stbt-debug/00004/level0-source_with_rois.png
stbt-debug/00004/level0-template.png
stbt-debug/00004/level1-source_matchtemplate.png
stbt-debug/00004/level1-source_matchtemplate_threshold.png
stbt-debug/00004/level1-source.png
stbt-debug/00004/level1-source_with_match.png
stbt-debug/00004/level1-source_with_rois.png
stbt-debug/00004/level1-template.png
stbt-debug/00004/level2-source_matchtemplate.png
stbt-debug/00004/level2-source_matchtemplate_threshold.png
stbt-debug/00004/level2-source.png
stbt-debug/00004/level2-source_with_match.png
stbt-debug/00004/level2-source_with_rois.png
stbt-debug/00004/level2-template.png
stbt-debug/00004/match0-confirm-absdiff.png
stbt-debug/00004/match0-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match0-confirm-absdiff_threshold.png
stbt-debug/00004/match0-confirm-source_roi_gray.png
stbt-debug/00004/match0-confirm-source_roi.png
stbt-debug/00004/match0-confirm-template_gray.png
stbt-debug/00004/match0-heatmap.png
stbt-debug/00004/match0-source_with_match.png
stbt-debug/00004/match1-confirm-absdiff.png
stbt-debug/00004/match1-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match1-confirm-absdiff_threshold.png
stbt-debug/00004/match1-confirm-source_roi_gray.png
stbt-debug/00004/match1-confirm-source_roi.png
stbt-debug/00004/match1-confirm-template_gray.png
stbt-debug/00004/match1-heatmap.png
stbt-debug/00004/match1-source_with_match.png
stbt-debug/00004/match2-confirm-absdiff.png
stbt-debug/00004/match2-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match2-confirm-absdiff_threshold.png
stbt-debug/00004/match2-confirm-source_roi_gray.png
stbt-debug/00004/match2-confirm-source_roi.png
stbt-debug/00004/match2-confirm-template_gray.png
stbt-debug/00004/match2-heatmap.png
stbt-debug/00004/match2-source_with_match.png
stbt-debug/00004/match3-confirm-absdiff.png
stbt-debug/00004/match3-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match3-confirm-absdiff_threshold.png
stbt-debug/00004/match3-confirm-source_roi_gray.png
stbt-debug/00004/match3-confirm-source_roi.png
stbt-debug/00004/match3-confirm-template_gray.png
stbt-debug/00004/match3-heatmap.png
stbt-debug/00004/match3-source_with_match.png
stbt-debug/00004/match4-confirm-absdiff.png
stbt-debug/00004/match4-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match4-confirm-absdiff_threshold.png
stbt-debug/00004/match4-confirm-source_roi_gray.png
stbt-debug/00004/match4-confirm-source_roi.png
stbt-debug/00004/match4-confirm-template_gray.png
stbt-debug/00004/match4-heatmap.png
stbt-debug/00004/match4-source_with_match.png
stbt-debug/00004/match5-confirm-absdiff.png
stbt-debug/00004/match5-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match5-confirm-absdiff_threshold.png
stbt-debug/00004/match5-confirm-source_roi_gray.png
stbt-debug/00004/match5-confirm-source_roi.png
stbt-debug/00004/match5-confirm-template_gray.png
stbt-debug/00004/match5-heatmap.png
stbt-debug/00004/match5-source_with_match.png
stbt-debug/00004/match6-confirm-absdiff.png
stbt-debug/00004/match6-confirm-absdiff_threshold_erode.png
stbt-debug/00004/match6-confirm-absdiff_threshold.png
stbt-debug/00004/match6-confirm-source_roi_gray.png
stbt-debug/00004/match6-confirm-source_roi.png
stbt-debug/00004/match6-confirm-template_gray.png
stbt-debug/00004/match6-heatmap.png
stbt-debug/00004/match6-source_with_match.png
stbt-debug/00004/source.png
stbt-debug/00004/source_with_matches.png
stbt-debug/00004/template.png
""")
assert_expected("stbt-debug-expected-output/match")
def test_sponge_that_new_data_end_up_in_file():
with scoped_curdir():
with _sponge('hello') as f:
f.write('hello')
assert open('hello').read() == 'hello'