Updated unit test_queries

Make codacy happy
TRON-Bioinformatics · Oct 31, 2023 · 9e5f130 · 9e5f130
1 parent 8c75c7e
commit 9e5f130
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/covigator/dashboard/tabs/overview.py b/covigator/dashboard/tabs/overview.py
@@ -1,6 +1,5 @@
 import dash_bootstrap_components as dbc
 from dash import dcc
-from dash import html
 from dash import Input, Output, State, html
 from covigator.database.queries import Queries
 from covigator.dashboard.tabs import COLOR_STATUS, NEWS_PATTERN

diff --git a/covigator/tests/unit_tests/test_queries.py b/covigator/tests/unit_tests/test_queries.py
@@ -306,6 +306,10 @@ def test_get_news(self):
              NewsSection(message_text="bla3", message_type=NewsType.RELEASE.name),
              NewsSection(message_text="bla4", message_type=NewsType.RELEASE.name)]
         self.session.add_all(test_news_section)
+        self.session.commit()
+
         news = self.queries.get_top_news()
         self.assertEqual(news.shape[0], 3)
+        # Check that most recent news item returned as top element
+        self.assertTrue(news.message_text.loc[0] == "bla4")