allow lower case/spaces, totalpower split vector

Author: amzhao16

pygridsim/results.py

@@ -8,20 +8,25 @@
 
 
 def _query_solution(query):
-    match query:
-        case "Voltages":
+    query_fix = query.lower().replace(" ", "")
+    match query_fix:
+        case "voltages":
             bus_vmags = {}
             for bus_name, bus_vmag in zip(altdss.BusNames(), altdss.BusVMag()):
                 bus_vmags[bus_name] = float(bus_vmag)
             return bus_vmags
-        case "Losses":
+        case "losses":
             vector_losses = altdss.Losses()
             losses = {}
             losses["Active Power Loss"] = vector_losses.real
             losses["Reactive Power Loss"] = vector_losses.imag
             return losses
-        case "TotalPower":
-            return altdss.TotalPower()
+        case "totalpower":
+            vector_power = altdss.TotalPower()
+            power = {}
+            power["Active Power"] = vector_power.real
+            power["Reactive Power"] = vector_power.imag
+            return power
         case _:
             return "Invalid"
 

tests/test_circuit.py

@@ -162,6 +162,16 @@ def test_011_configs(self):
         with self.assertRaises(Exception):
             circuit.update_source(source_type=SourceType.TURBINE)
 
+    def test_012_all_results(self):
+        circuit = PyGridSim()
+        circuit.update_source()
+        circuit.add_load_nodes()
+        circuit.add_generators(num=2, gen_type="small")
+        circuit.add_lines([("source", "load0"), ("generator0", "load0")])
+        circuit.solve()
+        # Should be flexible with capitalization, spaces
+        print(circuit.results(["Voltages", "losses", "Total Power"]))
+
 
 class TestCustomizedCircuit(unittest.TestCase):
     """