add type checking

Author: amzhao16

pygridsim/__pycache__/core.cpython-312.pyc

@@ -22,7 +22,7 @@ def __init__(self):
 		altdss.ClearAll()
 		altdss('new circuit.MyCircuit')
 
-	def add_load_nodes(self, load_type: str = "house", params = {}, num = 1):
+	def add_load_nodes(self, load_type: str = "house", params = {}, num: int = 1):
 		"""
 		When the user wants to manually add nodes, or make nodes with varying parameters.
 
@@ -55,7 +55,7 @@ def update_source(self, source_type: str = "turbine", params = {}):
 		"""
 		return make_source_node(params, source_type)
 
-	def add_PVSystem(self, load_nodes, params = {}, num_panels = 1):
+	def add_PVSystem(self, load_nodes: list[str], params = {}, num_panels: int = 1):
 		"""
 		Specify a list of load nodes to add a PVsystem ("solar panel") to.
 
@@ -74,7 +74,7 @@ def add_PVSystem(self, load_nodes, params = {}, num_panels = 1):
 			self.num_pv += 1
 		return PV_nodes
 
-	def add_generator(self, num, gen_type: GeneratorType = GeneratorType.SMALL, params = {}):
+	def add_generator(self, num: int = 1, gen_type: GeneratorType = GeneratorType.SMALL, params = {}):
 		"""
 		Specify parameters for a generator to add to the circuit
 
@@ -92,7 +92,7 @@ def add_generator(self, num, gen_type: GeneratorType = GeneratorType.SMALL, para
 		return generators
 
 
-	def add_lines(self, connections, line_type: str = "lv", params = {}, transformer = True):
+	def add_lines(self, connections: list[tuple], line_type: str = "lv", params = {}, transformer: bool = True):
 		"""
 		Specify all lines that the user wants to add. If redundant lines, doesn't add anything
 
@@ -104,7 +104,7 @@ def add_lines(self, connections, line_type: str = "lv", params = {}, transformer
 			make_line(src, dst, line_type, self.num_lines, params, transformer)
 			self.num_lines += 1
 
-	def view_load_nodes(self, indices = []):
+	def view_load_nodes(self, indices: list = []):
 		"""
 		View load nodes (what their parameters are) at the given indices.
 
@@ -151,7 +151,7 @@ def solve(self):
 		"""
 		altdss.Solution.Solve()
 
-	def results(self, queries, export_path = ""):
+	def results(self, queries: list[str], export_path = ""):
 		"""
 		Allow the user to query for many results at once instead of learning how to manually query
 

pygridsim/core.py

@@ -1,10 +1,10 @@
 {
     "Voltages": {
-        "source": 2077.428822659733,
-        "load0": 217.52249209519556
+        "source": 1391.2908459682467,
+        "load0": 202.6090367072441
     },
     "Losses": {
-        "Active Power Loss": 284917.58902059204,
-        "Reactive Power Loss": 592389.8824783975
+        "Active Power Loss": 78708.71838388889,
+        "Reactive Power Loss": 163597.67777927459
     }
 }