Update test_queries.py

TRON-Bioinformatics · Oct 26, 2023 · fc568f2 · fc568f2
1 parent b2f03c7
commit fc568f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/covigator/tests/unit_tests/test_queries.py b/covigator/tests/unit_tests/test_queries.py
@@ -298,5 +298,5 @@ def test_get_news(self):
         self.session.commit()
         news = self.queries.get_top_news()
         self.assertGreater(news.shape[0], 0)
-        self.assertTrue(news.message_text == "bla")
-        self.assertTrue(news.message_type == NewsType.RELEASE.name)
+        self.assertTrue(news.message_text.loc[0] == "bla")
+        self.assertTrue(news.message_type.loc[0] == NewsType.RELEASE.name)