add tests for new data functions

datamol-io · Jul 27, 2023 · be52130 · be52130
1 parent a3163aa
commit be52130
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_data.py b/tests/test_data.py
@@ -38,3 +38,15 @@ def test_solubility():
         "smiles",
         "split",
     ]
+
+
+def test_chembl_drugs():
+    data = dm.data.chembl_drugs()
+    assert data.shape == (1935, 1)
+    assert list(data.columns) == ["smiles"]
+
+
+def test_chembl_samples():
+    data = dm.data.chembl_samples()
+    assert data.shape == (2000, 1)
+    assert list(data.columns) == ["smiles"]