biology function test refactor

szuckerman · Jan 29, 2019 · 137a754 · 137a754
1 parent f394510
commit 137a754
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 27 deletions.
diff --git a/janitor/testing_utils/fixtures.py b/janitor/testing_utils/fixtures.py
@@ -1,6 +1,10 @@
 import numpy as np
 import pandas as pd
 import pytest
+from pathlib import Path
+
+
+TEST_DATA_DIR = Path(__file__).resolve().parent.parent.parent / "tests" / "test_data"
 
 
 @pytest.fixture
@@ -74,3 +78,9 @@ def missingdata_df():
     }
     df = pd.DataFrame(data)
     return df
+
+
+@pytest.fixture
+def biodf():
+    df = pd.read_csv(TEST_DATA_DIR / "sequences.tsv", sep="\t").clean_names()
+    return df
diff --git a/tests/biology/test_join_fasta.py b/tests/biology/test_join_fasta.py
@@ -0,0 +1,13 @@
+import janitor.biology
+import pytest
+
+from janitor.testing_utils.fixtures import TEST_DATA_DIR, biodf
+
+
+@pytest.mark.biology
+def test_join_fasta(biodf):
+    df = biodf.join_fasta(
+        TEST_DATA_DIR / "sequences.fasta", "sequence_accession", "sequence"
+    )
+
+    assert "sequence" in df.columns
diff --git a/tests/test_biology.py b/tests/test_biology.py