Skip to content

Commit

Permalink
tests: Add test cases for new functions
Browse files Browse the repository at this point in the history
  • Loading branch information
jjjermiah committed Feb 18, 2024
1 parent 51f19e8 commit 9fc3c36
Showing 1 changed file with 37 additions and 0 deletions.
37 changes: 37 additions & 0 deletions tests/test_nbia_series_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,28 @@ def nbia_series(nbia_client):
return series_df


@pytest.fixture(scope="session")
def getSeriesMetadata(nbia_client):
series = "1.3.6.1.4.1.14519.5.2.1.9203.4004.652695091345533290618011349477"
series_metadata = nbia_client.getSeriesMetadata(
SeriesInstanceUID=series, return_type="dataframe"
)
return series_metadata


@pytest.fixture(scope="session")
def getNewSeries_metadata(nbia_client):
Date = "01/01/2024"
new = nbia_client.getNewSeries(Date=Date, return_type="dataframe")
return new


@pytest.fixture(scope="session")
def getDICOMTags(nbia_client):
series = "1.3.6.1.4.1.14519.5.2.1.9203.4004.652695091345533290618011349477"
return nbia_client.getDICOMTags(SeriesInstanceUID=series, return_type="dataframe")


def test_tcga_studies(tcga_studies):

assert isinstance(tcga_studies, list)
Expand Down Expand Up @@ -102,3 +124,18 @@ def test_getSeries(nbia_series):
"SeriesNumber",
]
assert all(col in nbia_series.columns for col in expected_cols)


def test_getSeriesMetadata(getSeriesMetadata):
assert isinstance(getSeriesMetadata, pd.DataFrame)
assert len(getSeriesMetadata) == 1


def test_getNewSeries(getNewSeries_metadata):
assert isinstance(getNewSeries_metadata, pd.DataFrame)
assert len(getNewSeries_metadata) > 1


def test_getDICOMTags(getDICOMTags):
assert isinstance(getDICOMTags, pd.DataFrame)
assert len(getDICOMTags) > 1

0 comments on commit 9fc3c36

Please sign in to comment.