fix: add missing branch, dcline, sub comparisons

Breakthrough-Energy · May 19, 2020 · 2ff72c6 · 2ff72c6
1 parent 7fadc6c
commit 2ff72c6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/powersimdata/input/grid.py b/powersimdata/input/grid.py
@@ -119,6 +119,15 @@ def _univ_eq(ref, test):
             other_df = other.plant.drop(excluded_cols, axis=1)
             _univ_eq(self_df, other_df)
 
+            # compare branch
+            _univ_eq(self.branch, other.branch)
+
+            # compare dcline
+            _univ_eq(self.dcline, other.dcline)
+
+            # compare sub
+            _univ_eq(self.sub, other.sub)
+
             # check grid helper function equality
             _univ_eq(self.type2color, other.type2color)
             _univ_eq(self.id2type, other.id2type)