Spaces:
Sleeping
Sleeping
File size: 917 Bytes
458fa79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import unittest
from gridgent.agents.intent import IntentAgent
class TestIntentAgent(unittest.TestCase):
def setUp(self):
self.agent = IntentAgent()
def test_hosting_capacity_intent(self):
info = self.agent.classify("What happens on feeder F2 if we add 5 MW of rooftop PV?")
self.assertEqual(info["intent"], "hosting_capacity")
self.assertEqual(info["feeder"], "F2")
self.assertAlmostEqual(info["added_pv_mw"], 5.0, places=3)
def test_simulation_intent_load_growth(self):
info = self.agent.classify("Simulate load growth of 3 MW on feeder F1")
self.assertEqual(info["intent"], "simulation")
self.assertAlmostEqual(info["added_load_mw"], 3.0, places=3)
def test_unknown_for_smalltalk(self):
info = self.agent.classify("hi")
self.assertEqual(info["intent"], "unknown")
if __name__ == "__main__":
unittest.main()
|