TST Add failing test for very short peptides

BigDataBiology · May 19, 2021 · 4267c99 · 4267c99
1 parent f2d21db
commit 4267c99
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/macrel/tests/data/very_short.faa b/macrel/tests/data/very_short.faa
@@ -0,0 +1,2 @@
+>short
+AA
diff --git a/macrel/tests/test_predict.py b/macrel/tests/test_predict.py
@@ -1,6 +1,7 @@
 import numpy as np
 from macrel import AMP_features, AMP_predict
 from macrel.main import data_file
+from os import path
 
 def test_predict():
     fs = AMP_features.features('tests/peptides/expep.faa.gz')
@@ -12,3 +13,10 @@ def test_predict():
                         fs, keep_negatives=True)
     assert len(fsp) < len(fsn)
     assert not np.all(fsn.is_AMP)
+
+def test_predict_very_short():
+    fs = AMP_features.features(
+            path.join(path.dirname(__file__),
+                    'data',
+                    'very_short.faa')) 
+    assert len(fs) == 1