Add decoy_prefix and decoy_suffix to protxml tests. Version 3.5

levitsky · May 28, 2018 · 2747b67 · 2747b67
1 parent 31dd8dd
commit 2747b67
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-3.5b6
+3.5
diff --git a/tests/test_protxml.py b/tests/test_protxml.py
@@ -41,16 +41,30 @@ def test_qvalues(self):
         q = protxml.qvalues(self.path, **self._kw)
         self.assertEqual(list(q['q']), [0, 1])
 
+    def test_qvalues_prefix(self):
+        q = protxml.qvalues(self.path, decoy_prefix='DECO', **self._kw)
+        self.assertEqual(list(q['q']), [0, 1])
+
     def test_df(self):
         df = protxml.DataFrame(self.path)
         self.assertEqual(df.shape, (2, 15))
 
     def test_filter_df(self):
         kw = self._kw.copy()
         del kw['full_output']
-        kw['key'] = 'probability'
+        del kw['key']
         fdf = protxml.filter_df(self.path, **kw)
         self.assertEqual(fdf.shape, (2, 17))
 
+    def test_filter_df_suffix(self):
+        kw = self._kw.copy()
+        del kw['full_output']
+        del kw['key']
+        kw['remove_decoy'] = True
+        df = protxml.DataFrame(self.path)
+        df['protein_name'] = df.protein_name.str.replace(r'DECOY_(.*)', r'\1_SUF')
+        fdf = protxml.filter_df(df, decoy_suffix='_SUF', **kw)
+        self.assertEqual(fdf.shape, (1, 17))
+
 if __name__ == '__main__':
     unittest.main()