Updates tests
This commit is contained in:
Родитель
53e181d2a9
Коммит
6790eb72ac
|
@ -19,8 +19,8 @@ class MockModel(object):
|
||||||
def test_init(monkeypatch):
|
def test_init(monkeypatch):
|
||||||
""" Tests model initialisation
|
""" Tests model initialisation
|
||||||
"""
|
"""
|
||||||
monkeypatch.setattr(driver, 'LABEL_FILE', '../synset.txt')
|
monkeypatch.setattr(driver, 'LABEL_FILE', 'synset.txt')
|
||||||
monkeypatch.setattr(driver, 'MODEL_FILE', '../ResNet_152.model')
|
monkeypatch.setattr(driver, 'MODEL_FILE', 'ResNet_152.model')
|
||||||
init()
|
init()
|
||||||
assert driver.trainedModel != None
|
assert driver.trainedModel != None
|
||||||
|
|
||||||
|
|
Загрузка…
Ссылка в новой задаче