diff --git a/covigator/tests/unit_tests/test_queries.py b/covigator/tests/unit_tests/test_queries.py index b1a6494..59a87da 100644 --- 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)