diff --git a/tests/test_parsers.py b/tests/test_parsers.py index a387e5c2..c8ae83b0 100644 --- a/tests/test_parsers.py +++ b/tests/test_parsers.py @@ -292,6 +292,32 @@ def test_read_sssom_table(self): """Test read SSSOM method to validate import of all columns.""" input_path = os.path.join(test_data_dir, "basic3.tsv") msdf = parse_sssom_table(input_path) + self.assertIn("comment", set(msdf.df.columns)) + + # note that the subject_category and object_category get re-organized + self.assertEqual( + [ + "subject_id", + "subject_label", + "subject_category", + "predicate_id", + "predicate_modifier", + "object_id", + "object_label", + "object_category", + "mapping_justification", + "subject_source", + "object_source", + "mapping_tool", + "confidence", + "subject_match_field", + "object_match_field", + "match_string", + "comment", + ], + list(msdf.df.columns), + ) + imported_df = pd.read_csv(input_path, comment="#", sep="\t").fillna("") imported_df = sort_df_rows_columns(imported_df) msdf.df = sort_df_rows_columns(msdf.df)