From 38013d2164c44cd86d17086f16832e2b76ce5ea0 Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Tue, 30 Jan 2024 13:47:47 -0500 Subject: [PATCH] cleanup: `test_datasets` --- tests/test_datasets.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index e21e50d..b91ded2 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -1,5 +1,3 @@ -from typing import Any - import pytest from requests import ConnectionError @@ -22,16 +20,13 @@ def test_dataset_constructor() -> None: Datasets(db, metadata_file=bad_metadata_url) -def test_list_datasets(capfd: Any) -> None: +def test_list_datasets() -> None: datasets = Datasets(db, metadata_file=test_metadata_url).list_datasets() assert type(datasets) is list assert "TEST" in datasets - out, _ = capfd.readouterr() - assert "TEST" in out - -def test_dataset_info(capfd: Any) -> None: +def test_dataset_info() -> None: with pytest.raises(ValueError): Datasets(db).dataset_info("invalid") @@ -45,9 +40,6 @@ def test_dataset_info(capfd: Any) -> None: assert type(dataset) is dict assert dataset["file_type"] == "jsonl" - out, _ = capfd.readouterr() - assert len(out.replace("\n", "")) > 0 - def test_load_json() -> None: cleanup_collections()