Improve testing (#12)

* Improve testing

* Bump version

README.md

@@ -1,6 +1,6 @@
 # <img src="https://raw.githubusercontent.com/vsakkas/bard.py/master/images/logo.svg?token=GHSAT0AAAAAAB7MEK465TODCKRPHN3YQY54ZKGUN4Q" width="28px" /> Bard.py
 
-[![Latest Release](https://img.shields.io/github/v/release/vsakkas/bard.py.svg)](https://github.com/vsakkas/bard.py/releases/tag/v0.2.1)
+[![Latest Release](https://img.shields.io/github/v/release/vsakkas/bard.py.svg)](https://github.com/vsakkas/bard.py/releases/tag/v0.2.2)
 [![Python](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
 [![MIT License](https://img.shields.io/badge/license-MIT-blue)](https://github.com/vsakkas/bard.py/blob/master/LICENSE)
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bard-py"
-version = "0.2.1"
+version = "0.2.2"
 description = "Python Client for Bard."
 authors = ["vsakkas <[email protected]>"]
 license = "MIT"

tests/test_ask.py

@@ -1,12 +1,28 @@
 import pytest
+from thefuzz import fuzz
 
 from bard import BardClient
 
 
 @pytest.mark.asyncio
 async def test_ask() -> None:
+    expected_responses = [
+        "Hello there! How can I help you today?",
+        "Hello there! How can I help you today? [Image of Bard AI Chatbot]",
+        "Hello there! How can I help you today? [Image of Bard, a large language model from Google AI]",
+        "Hello there! How can I help you today? [Image of a friendly robot with a warm smile]",
+    ]
+
     async with BardClient() as bard:
-        _ = await bard.ask("Tell me a joke")
+        response = await bard.ask("Hello, Bard!")
+
+        score = 0
+        for expected_response in expected_responses:
+            score = fuzz.token_sort_ratio(response, expected_response)
+            if score >= 80:
+                return True
+
+        assert False, f"Unexpected response: {response}, match score: {score}"
 
 
 @pytest.mark.asyncio