1 files changed, 34 insertions, 0 deletions
diff --git a/benchmark/agbenchmark/challenges/library/ethereum/check_price/artifacts_in/test.py b/benchmark/agbenchmark/challenges/library/ethereum/check_price/artifacts_in/test.py
new file mode 100644
index 000000000..76a2e299b
--- /dev/null
+++ b/benchmark/agbenchmark/challenges/library/ethereum/check_price/artifacts_in/test.py
@@ -0,0 +1,34 @@
+import re
+
+from sample_code import get_ethereum_price
+
+
+def test_get_ethereum_price() -> None:
+    # Read the Ethereum price from the file
+    with open("eth_price.txt", "r") as file:
+        eth_price = file.read().strip()
+
+    # Validate that the eth price is all digits
+    pattern = r"^\d+$"
+    matches = re.match(pattern, eth_price) is not None
+    assert (
+        matches
+    ), f"AssertionError: Ethereum price should be all digits, but got {eth_price}"
+
+    # Get the current price of Ethereum
+    real_eth_price = get_ethereum_price()
+
+    # Convert the eth price to a numerical value for comparison
+    eth_price_value = float(eth_price)
+    real_eth_price_value = float(real_eth_price)
+
+    # Check if the eth price is within $50 of the actual Ethereum price
+    assert (
+        abs(real_eth_price_value - eth_price_value) <= 50
+    ), f"AssertionError: Ethereum price is not within $50 of the actual Ethereum price (Provided price: ${eth_price}, Real price: ${real_eth_price})"
+
+    print("Matches")
+
+
+if __name__ == "__main__":
+    test_get_ethereum_price()