Adding ark_model unit tests

tests/test_discovery_ark_model.py

@@ -0,0 +1,60 @@
+""" discovery/ark_model.py tests """
+import unittest
+from unittest import mock
+from io import StringIO
+import pandas as pd
+
+# pylint: disable=unused-import
+from gamestonk_terminal.test_helper import (  # noqa: F401
+    parameterize_from_file,
+    pytest_generate_tests,
+)
+
+from gamestonk_terminal.discovery.ark_model import get_ark_orders, add_order_total
+
+assertions = unittest.TestCase("__init__")
+
+
+class TestDiscoveryFidelityModel:
+    @mock.patch("gamestonk_terminal.discovery.ark_model.requests")
+    @parameterize_from_file(
+        "test_get_ark_orders",
+        "../tests/data/discovery_ark_model.yaml",
+    )
+    # pylint: disable=unused-argument
+    def test_get_ark_orders(
+        self,
+        mock_requests_get,
+        mock_ark_orders_html,
+        expected_orders,
+    ):
+        mock_requests_get.get().text = mock_ark_orders_html
+        ret = get_ark_orders()
+
+        assertions.assertEqual.__self__.maxDiff = None
+        assertions.assertEqual(
+            ret.to_csv().replace("\r\n", "\n"), expected_orders.replace("\r\n", "\n")
+        )
+
+    @mock.patch("gamestonk_terminal.discovery.ark_model.yf.download")
+    @parameterize_from_file(
+        "test_add_order_total",
+        "../tests/data/discovery_ark_model.yaml",
+    )
+    # pylint: disable=unused-argument
+    def test_add_order_total(
+        self, mock_yf_download, mock_df_orders, mock_yf_result, expected_orders
+    ):
+        df_orders = pd.read_csv(
+            StringIO(mock_df_orders), header=0, index_col=0, parse_dates=["date"]
+        )
+        mock_yf_download.return_value = pd.read_csv(
+            StringIO(mock_yf_result), header=[0, 1], index_col=0, parse_dates=True
+        )
+
+        ret = add_order_total(df_orders)
+
+        assertions.assertEqual.__self__.maxDiff = None
+        assertions.assertEqual(
+            ret.to_csv().replace("\r\n", "\n"), expected_orders.replace("\r\n", "\n")
+        )