diff --git a/tests/test_new_query_functionality.py b/tests/test_new_query_functionality.py index e7f8250..d7495e0 100644 --- a/tests/test_new_query_functionality.py +++ b/tests/test_new_query_functionality.py @@ -1,6 +1,8 @@ import pytest +from nbiatoolkit import nbia from src.nbiatoolkit import NBIAClient +from src.nbiatoolkit.auth import OAuth2 from src.nbiatoolkit.utils import * import pandas as pd @@ -42,6 +44,28 @@ def nbia_patients_df(nbia_client): return nbia_patients_df +@pytest.fixture(scope="session") +def nbia_collections_by_colMoldality(nbia_client): + nbia_collections_by_colMoldality = nbia_client.getPatientsByCollectionAndModality( + Collection="TCGA-KIRC", Modality="MR" + ) + return nbia_collections_by_colMoldality + + +def test_nbia_properties(nbia_client): + assert isinstance(nbia_client.OAuth_client, OAuth2) + assert isinstance(nbia_client.headers, dict) + assert "Authorization" in nbia_client.headers.keys() + assert "Content-Type" in nbia_client.headers.keys() + assert nbia_client.headers["Content-Type"] == "application/json" + + assert nbia_client.base_url == NBIA_ENDPOINTS.NBIA + nbia_client.base_url = NBIA_ENDPOINTS.NLST + assert nbia_client.base_url == NBIA_ENDPOINTS.NLST + + assert nbia_client.logger is not None + + def test_tcga_collection(tcga_collections): tcga_collections_df, tcga_collections1 = tcga_collections assert isinstance(tcga_collections1, list) @@ -124,3 +148,8 @@ def test_failed_getNewPatients(nbia_client): nbia_client.getNewPatients( collection="TCGA", return_type="dataframe", Date="2022/12/06" ) + + +def test_nbia_collections_by_colMoldality(nbia_collections_by_colMoldality): + assert isinstance(nbia_collections_by_colMoldality, list) + assert len(nbia_collections_by_colMoldality) > 1