From fc568f238cfc0529e5de8cac427d5797d428e218 Mon Sep 17 00:00:00 2001
From: Jo Hausmann <35309108+johausmann@users.noreply.github.com>
Date: Thu, 26 Oct 2023 16:12:20 +0200
Subject: [PATCH] Update test_queries.py

---
 covigator/tests/unit_tests/test_queries.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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)