def test_locate_one_image():
dirs = [p for p in os.listdir('testdata') if p.startswith('oneimg-')]
for d in dirs:
folder = os.path.join('testdata', d)
EXPECT_FILE = os.path.join(folder, 'expect.txt')
TRAIN_FILE = os.path.join(folder, 'train.png')
if not os.path.exists(TRAIN_FILE):
TRAIN_FILE = os.path.join(folder, 'train.jpg')
assert os.path.exists(TRAIN_FILE)
#
# format: <filename> <left-up> <right-bottom>
# eg: q1.png 100 200 40 20
#
if os.path.exists(EXPECT_FILE):
for line in open(EXPECT_FILE):
line = line.strip()
if not line or line.startswith('#'):
continue
name, extra = string.split(line, maxsplit=1)
query_file = os.path.join(folder, name)
print 'TEST EXISTS:', query_file, extra
x0, y0, x1, y1 = map(int, extra.split())
assert x0 < x1
assert y0 < y1
point = image.locate_one_image(TRAIN_FILE, query_file)
assert point != None
(x, y) = point
assert x0 < x < x1
assert y0 < y < y1
NOEXISTS_FILE = os.path.join(folder, 'noexists.txt')
if os.path.exists(NOEXISTS_FILE):
for line in open(NOEXISTS_FILE):
line = line.strip()
if not line or line.startswith('#'):
continue
filename = os.path.join(folder, line)
print 'TEST NOT EXISTS:', filename
point = image.locate_one_image(TRAIN_FILE, filename)
assert point == None
def test_locate_one_image():
dirs = [p for p in os.listdir('testdata') if p.startswith('oneimg-')]
for d in dirs:
folder = os.path.join('testdata', d)
EXPECT_FILE = os.path.join(folder, 'expect.txt')
TRAIN_FILE = os.path.join(folder, 'train.png')
if not os.path.exists(TRAIN_FILE):
TRAIN_FILE = os.path.join(folder, 'train.jpg')
assert os.path.exists(TRAIN_FILE)
#
# format: <filename> <left-up> <right-bottom>
# eg: q1.png 100 200 40 20
#
if os.path.exists(EXPECT_FILE):
for line in open(EXPECT_FILE):
line = line.strip()
if not line or line.startswith('#'):
continue
name, extra = string.split(line, maxsplit=1)
query_file = os.path.join(folder, name)
print 'TEST EXISTS:', query_file, extra
x0, y0, x1, y1 = map(int, extra.split())
assert x0 < x1
assert y0 < y1
point = image.locate_one_image(TRAIN_FILE, query_file)
assert point != None
(x, y) = point
assert x0 < x < x1
assert y0 < y < y1
NOEXISTS_FILE = os.path.join(folder, 'noexists.txt')
if os.path.exists(NOEXISTS_FILE):
for line in open(NOEXISTS_FILE):
line = line.strip()
if not line or line.startswith('#'):
continue
filename = os.path.join(folder, line)
print 'TEST NOT EXISTS:', filename
point = image.locate_one_image(TRAIN_FILE, filename)
assert point == None
def test_image_locate_file_not_exists():
with pytest.raises(IOError):
image.locate_one_image('yy.png', 'zz.png')
with pytest.raises(IOError):
image.locate_one_image('testdata/yy.png', 'zz.png')
def test_image_locate_file_not_exists():
with pytest.raises(IOError):
image.locate_one_image('yy.png', 'zz.png')
with pytest.raises(IOError):
image.locate_one_image('testdata/yy.png', 'zz.png')
