Increased test timeout for scneario in which the model should be downloaded

This commit is contained in:
Nadav Bar 2017-02-28 15:06:46 +02:00
Родитель ce69d92135
Коммит bd1305cfa2
1 изменённых файлов: 7 добавлений и 3 удалений

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

@ -12,17 +12,21 @@ import pytest
abs_path = os.path.dirname(os.path.abspath(__file__))
notebook = os.path.join(abs_path, "..", "..", "..", "..", "Examples","Image","Detection","FastRCNN", "CNTK_FastRCNN_Eval.ipynb")
# Skipping test for python 2.7 since FastRCNN implementatio does not support 2.7 at the moment
# setting a large timeout in case we need to download the Fast-RCNN pretrained model
notebook_timeoutSeconds = 1200
# Skipping test for python 2.7 since Fast-RCNN implementation does not support 2.7 at the moment
@pytest.mark.skipif(sys.version_info < (3,4),
reason="requires python3.4")
reason="requires python 3.4")
def test_cntk_fastrcnn_eval_noErrors(nb):
errors = [output for cell in nb.cells if 'outputs' in cell
for output in cell['outputs'] if output.output_type == "error"]
assert errors == []
# Skipping test for python 2.7 since Fast-RCNN implementation does not support 2.7 at the moment
@pytest.mark.skipif(sys.version_info < (3,4),
reason="requires python3.4")
reason="requires python 3.4")
def test_cntk_fastrcnn_eval_evalCorrect(nb):
testCells = [cell for cell in nb.cells
if cell.cell_type == 'code' and