This commit is contained in:
MSalvaris 2017-11-22 16:51:21 +00:00
Родитель 53e181d2a9
Коммит 6790eb72ac
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -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