class TestBlocking: def setup(self): self.rs = RandomStuff() def test_invalid_joke_error(self): with pytest.raises(RuntimeError) as exc: self.rs.get_joke("deva") assert "deva" in str(exc) def test_invalid_image_error(self): with pytest.raises(RuntimeError) as exc: self.rs.get_image("awww") assert "awww" in str(exc) def test_joke_response(self): response = self.rs.get_joke("dev") assert isinstance(response, dict) assert set(response.keys()).issuperset(joke_response_keys) def test_ai_response(self): response = self.rs.get_ai_response("Hello!") assert isinstance(response, str) def test_image_reponse(self): response = self.rs.get_image("aww") assert isinstance(response, str) def teardown(self): self.rs.close()
class TestAsync: def setup(self): self.rs = RandomStuff(async_mode=True) self.loop = get_event_loop() self.run = self.loop.run_until_complete def test_joke_response(self): response = self.run(self.rs.get_joke("dev")) assert isinstance(response, dict) assert set(response.keys()).issuperset(joke_response_keys) def test_image_response(self): response = self.run(self.rs.get_image("aww")) assert isinstance(response, str) def test_ai_response(self): response = self.run(self.rs.get_ai_response("Hello!")) assert isinstance(response, str)
def post_message(): message = request.form['message'] rs = RandomStuff() response = rs.get_ai_response(message) return f"""<h1>You >> {message}<br> Spot >>> {response}<br></h1>"""
# import the module from prsaw import RandomStuff # initiate the object rs = RandomStuff() # get a response from an endpoint response = rs.get_ai_response("How are you?") print(response) # close the object once done (recommended) rs.close()
import discord, os from life import revive_stamina from prsaw import RandomStuff rs = RandomStuff() client = discord.Client(async_mode=True) res = rs.get_ai_response("Hi there!") @client.event async def on_ready(): print("Let's go, {0.user}!".format(client)) @client.event async def on_message(message): if client.user == message.author: return res = rs.get_ai_response(message.content) await message.reply(res) revive_stamina() client.run(os.getenv('TOKEN'))