Source code for evennia.contrib.tutorials.evadventure.tests.test_ai
"""
Test the ai module.
"""
from unittest.mock import Mock, patch
from evennia import create_object
from evennia.utils.test_resources import BaseEvenniaTest
from ..characters import EvAdventureCharacter
from ..npcs import EvAdventureMob
[docs]class TestAI(BaseEvenniaTest):
[docs] def setUp(self):
super().setUp()
self.npc = create_object(EvAdventureMob, key="Goblin", location=self.room1)
self.pc = create_object(EvAdventureCharacter, key="Player", location=self.room1)
[docs] def tearDown(self):
super().tearDown()
self.npc.delete()
[docs] @patch("evennia.contrib.tutorials.evadventure.ai.random.random")
@patch("evennia.contrib.tutorials.evadventure.ai.log_trace")
def test_ai_methods(self, mock_log_trace, mock_random):
self.assertEqual(self.npc.ai.get_state(), "idle")
self.npc.ai.set_state("roam")
self.assertEqual(self.npc.ai.get_state(), "roam")
self.assertEqual(self.npc.ai.get_targets(), [self.pc])
self.assertEqual(self.npc.ai.get_traversable_exits(), [self.exit])
probs = {"hold": 0.1, "combat": 0.5, "flee": 0.4}
mock_random.return_value = 0.3
self.assertEqual(self.npc.ai.random_probability(probs), "combat")
mock_random.return_value = 0.7
self.assertEqual(self.npc.ai.random_probability(probs), "flee")
mock_random.return_value = 0.95
self.assertEqual(self.npc.ai.random_probability(probs), "hold")
[docs] def test_ai_run(self):
self.npc.ai.set_state("roam")
self.assertEqual(self.npc.ai.get_state(), "roam")
self.npc.ai.run()
self.assertEqual(self.npc.ai.get_state(), "combat")