Fix Discount and Mercantile Gold (XAU) issue

jond01 · Oct 28, 2024 · 13f1b03 · 13f1b03
1 parent 2d467d6
commit 13f1b03
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/tests/test_discount.py b/tests/test_discount.py
@@ -63,13 +63,16 @@ def currencies_fixture() -> set[CurrencyCode]:
     }
 
 
+@pytest.mark.parametrize(
+    "dropped_currencies", [[CurrencyCode.XAU]]
+)  # There's an issue with Gold
 @pytest.mark.live
 def test_df(
-    df: pd.DataFrame, currencies: set[CurrencyCode], drop_ngn: bool = True
+    df: pd.DataFrame,
+    currencies: set[CurrencyCode],
+    dropped_currencies: list[CurrencyCode],
 ) -> None:
     assert set(df.index) == currencies
-    if drop_ngn:
-        # There's an issue with Nigerian naira
-        df = df.drop(index=CurrencyCode.NGN)
+    df = df.drop(index=dropped_currencies)
     for method in ("cash", "transfer"):
         assert (df[(method, "sell")] >= df[(method, "buy")]).all()
diff --git a/tests/test_mercantile.py b/tests/test_mercantile.py
@@ -49,6 +49,11 @@ def currencies_fixture() -> set[CurrencyCode]:
     }
 
 
+@pytest.mark.parametrize("dropped_currencies", [[CurrencyCode.XAU]])
 @pytest.mark.live
-def test_df(df: pd.DataFrame, currencies: set[CurrencyCode]) -> None:
-    _test_df_helper(df, currencies, drop_ngn=False)
+def test_df(
+    df: pd.DataFrame,
+    currencies: set[CurrencyCode],
+    dropped_currencies: list[CurrencyCode],
+) -> None:
+    _test_df_helper(df, currencies, dropped_currencies=dropped_currencies)