def test_find_version_none():
gold = "1"
items = []
with patch('mead.utils._listdir') as list_patch:
list_patch.return_value = items
res = find_model_version(None)
assert res == gold
def test_find_version_non_int_only():
gold = "1"
items = [rand_str for _ in range(random.randint(1, 3))]
with patch('mead.utils._listdir') as list_patch:
list_patch = items
res = find_model_version(None)
assert res == gold
def test_find_version():
gold = random.randint(5, 15)
items = list(map(str, range(gold)))
gold = str(gold)
with patch('mead.utils._listdir') as list_patch:
list_patch.return_value = items
res = find_model_version(None)
assert res == gold
def test_find_version_non_int():
gold = random.randint(5, 15)
items = list(map(str, random.sample(range(gold), gold // 2)))
items.append(str(gold - 1))
gold = str(gold)
items.insert(random.randint(0, len(items)), rand_str())
with patch('mead.utils._listdir') as list_patch:
list_patch.return_value = items
res = find_model_version(None)
assert res == gold