fix pandas warning

msaltnet · May 6, 2024 · 601b275 · 601b275
1 parent 02fb5c0
commit 601b275
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/unit_tests/analyzer_test.py b/tests/unit_tests/analyzer_test.py
@@ -880,16 +880,16 @@ def test_get_return_report_draw_graph_when_graph_filename_exist(
         self.assertEqual(mock_addplot.call_args_list[4][1]["type"], "scatter")
         self.assertEqual(mock_addplot.call_args_list[4][1]["markersize"], 50)
         self.assertEqual(mock_addplot.call_args_list[4][1]["marker"], ".")
-        self.assertEqual(mock_addplot.call_args_list[0][0][0][0], 5000)
-        self.assertEqual(mock_addplot.call_args_list[0][0][0][1], 5000)
-        self.assertEqual(mock_addplot.call_args_list[1][0][0][1], 6000.0)
-        self.assertEqual(mock_addplot.call_args_list[2][0][0][0], 500)
-        self.assertEqual(mock_addplot.call_args_list[2][0][0][1], 600)
-        self.assertEqual(mock_addplot.call_args_list[3][0][0][0], -65.248)
-        self.assertEqual(mock_addplot.call_args_list[3][0][0][1], -75.067)
-        self.assertEqual(mock_addplot.call_args_list[4][0][0][0], 5900)
-        self.assertEqual(mock_addplot.call_args_list[4][0][0][1], 9999)
-        self.assertEqual(mock_addplot.call_args_list[4][0][0][2], 8888)
+        self.assertEqual(mock_addplot.call_args_list[0][0][0].iloc[0], 5000)
+        self.assertEqual(mock_addplot.call_args_list[0][0][0].iloc[1], 5000)
+        self.assertEqual(mock_addplot.call_args_list[1][0][0].iloc[1], 6000.0)
+        self.assertEqual(mock_addplot.call_args_list[2][0][0].iloc[0], 500)
+        self.assertEqual(mock_addplot.call_args_list[2][0][0].iloc[1], 600)
+        self.assertEqual(mock_addplot.call_args_list[3][0][0].iloc[0], -65.248)
+        self.assertEqual(mock_addplot.call_args_list[3][0][0].iloc[1], -75.067)
+        self.assertEqual(mock_addplot.call_args_list[4][0][0].iloc[0], 5900)
+        self.assertEqual(mock_addplot.call_args_list[4][0][0].iloc[1], 9999)
+        self.assertEqual(mock_addplot.call_args_list[4][0][0].iloc[2], 8888)
 
     @patch("pandas.to_datetime")
     @patch("pandas.DataFrame")