From c5811ae2130f7ba89a2d930c87b5864795ba4522 Mon Sep 17 00:00:00 2001
From: KillianLucas <63927363+KillianLucas@users.noreply.github.com>
Date: Fri, 11 Aug 2023 04:18:56 +0000
Subject: [PATCH] Improved tests

---
 tests/test_interpreter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_interpreter.py b/tests/test_interpreter.py
index 8fd128846..af74a4062 100644
--- a/tests/test_interpreter.py
+++ b/tests/test_interpreter.py
@@ -11,7 +11,7 @@ def test_hello_world():
 def test_math():
     interpreter.reset()
     messages = interpreter.chat("""Please perform the calculation 27073*7397 then reply with just the integer answer with no commas or anything, nothing else.""", return_messages=True)
-    assert messages[-1] == {'role': 'assistant', 'content': '200258981'}
+    assert "200258981" in messages[-1]["content"]
 
 def test_delayed_exec():
     interpreter.reset()