mark smoke for two mind tests; updated pytest.ini

This commit is contained in:
Ubuntu 2020-11-08 15:27:29 +00:00
Родитель ddc474b226
Коммит c8511f8ea3
2 изменённых файлов: 4 добавлений и 0 удалений

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

@ -1,5 +1,7 @@
[pytest] [pytest]
markers = markers =
deeprec: test deeprec model
sequential: test sequential model
notebooks: mark a test as notebooks test notebooks: mark a test as notebooks test
smoke: mark a test as smoke test smoke: mark a test as smoke test
gpu: mark a test as gpu test gpu: mark a test as gpu test

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

@ -6,6 +6,7 @@ import os
import requests import requests
from reco_utils.dataset.mind import download_mind, extract_mind from reco_utils.dataset.mind import download_mind, extract_mind
@pytest.mark.smoke
@pytest.mark.parametrize("url, content_length, etag", @pytest.mark.parametrize("url, content_length, etag",
[("https://recodatasets.blob.core.windows.net/newsrec/MINDdemo_train.zip", [("https://recodatasets.blob.core.windows.net/newsrec/MINDdemo_train.zip",
"17372879", "0x8D82C63E386D09C"), "17372879", "0x8D82C63E386D09C"),
@ -34,6 +35,7 @@ def test_mind_url(url, content_length, etag):
assert url_headers["Content-Length"] == content_length assert url_headers["Content-Length"] == content_length
assert url_headers["ETag"] == etag assert url_headers["ETag"] == etag
@pytest.mark.smoke
@pytest.mark.parametrize("size",[("demo"),("small")]) @pytest.mark.parametrize("size",[("demo"),("small")])
def test_extract_mind(size,tmp): def test_extract_mind(size,tmp):
""" Test file download and extration for demo and small datasets """ """ Test file download and extration for demo and small datasets """