Increased test timeout for scneario in which the model should be downloaded
This commit is contained in:
Родитель
ce69d92135
Коммит
bd1305cfa2
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче