tests: add more mock responses

This commit is contained in:
2026-04-11 22:12:22 -04:00
parent a9faf300bf
commit b589b936e6
5 changed files with 4981 additions and 4 deletions
+4 -4
View File
@@ -3,8 +3,8 @@ import polars as pl
from pcdigitizer import Annotation, PubChemAPI
def test_source_df():
df = PubChemAPI.get_sources()
def test_source_df(mock_session):
df = PubChemAPI.get_sources(session=mock_session)
columns = df.columns
assert len(columns) == 23
@@ -18,9 +18,9 @@ def test_source_df():
assert source_count == 1
def test_source_annotations():
def test_source_annotations(mock_session):
annotations = PubChemAPI.get_source_annotations(
"Hazardous Substances Data Bank (HSDB)"
"Hazardous Substances Data Bank (HSDB)", session=mock_session
)
assert len(annotations["Compound"]) == 246