Fix enzyme and sushiswap tests

safe-global · Apr 21, 2022 · eed4379 · eed4379
1 parent 0f4b4f4
commit eed4379
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gnosis/eth/tests/test_oracles.py b/gnosis/eth/tests/test_oracles.py
@@ -27,6 +27,7 @@
 
 gno_token_mainnet_address = "0x6810e776880C02933D47DB1b9fc05908e5386b96"
 weth_token_mainnet_address = "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"
+wbtc_token_mainnet_address = "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599"
 
 dai_token_mainnet_address = "0x6B175474E89094C44Da98b954EedeAC495271d0F"
 usdt_token_mainnet_address = "0xdAC17F958D2ee523a2206206994597C13D831ec7"
@@ -214,9 +215,8 @@ def test_get_price(self):
         sushiswap_oracle = SushiswapOracle(ethereum_client)
 
         price = sushiswap_oracle.get_price(
-            gno_token_mainnet_address, weth_token_mainnet_address
+            wbtc_token_mainnet_address, weth_token_mainnet_address
         )
-        self.assertLess(price, 1)
         self.assertGreater(price, 0)
 
         # Test with 2 stablecoins
@@ -518,7 +518,7 @@ def test_get_underlying_tokens(self):
         self.assertEqual(underlying_token.address, mln_vault_underlying_token)
 
         underlying_tokens = enzyme_oracle.get_underlying_tokens(usf_fund_token_address)
-        self.assertEqual(len(underlying_tokens), 2)
+        self.assertEqual(len(underlying_tokens), 1)
         for underlying_token in underlying_tokens:
             self.assertIn(underlying_token.address, usf_fund_underlying_tokens)
             self.assertAlmostEqual(underlying_token.quantity, 0.5, delta=0.5)