actual single pred inverted
@@ -51,10 +51,10 @@ class TestKGTKFilter(unittest.TestCase):
self.assertEqual(len(df), 1)
def test_kgtk_filter_spo(self):
- df = self.df2.loc[self.df2['sub'] != 'Q65695069']
- cli_entry("kgtk", "filter", "-i", self.file_path2, "-o", f'{self.temp_dir}/Q65695069.tsv', "-p",
- "Q65695069;;", "--subj", "sub", "--pred", "pred", "--obj", "obj", "-v", "--invert")
+ df = self.df2.loc[self.df2['pred'] != 'P577']
+ cli_entry("kgtk", "filter", "-i", self.file_path2, "-o", f'{self.temp_dir}/P577.tsv', "-p",
+ ";P577;", "--subj", "sub", "--pred", "pred", "--obj", "obj", "-v", "--invert")
- df_r = pd.read_csv(f'{self.temp_dir}/Q65695069.tsv', sep='\t')
+ df_r = pd.read_csv(f'{self.temp_dir}/P577.tsv', sep='\t')
self.assertEqual(len(df_r), len(df))
