fix: check gencost column of storage

Breakthrough-Energy · May 18, 2020 · 7fadc6c · 7fadc6c
1 parent 92e676f
commit 7fadc6c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/powersimdata/input/grid.py b/powersimdata/input/grid.py
@@ -78,7 +78,6 @@ def _univ_eq(ref, test):
             err_msg = 'Unable to compare Grid & %s' % type(other).__name__
             raise NotImplementedError(err_msg)
 
-        # Check all AbstractGridField attributes
         try:
             # compare gencost
             # Comparing 'after' will fail if one Grid was linearized
@@ -87,8 +86,8 @@ def _univ_eq(ref, test):
             _univ_eq(self_data, other_data)
 
             # compare storage
-            self_storage_num = self.gencost
-            other_storage_num = other.gencost
+            self_storage_num = len(self.storage['gencost'])
+            other_storage_num = len(other.storage['gencost'])
             if self_storage_num == 0:
                 assert other_storage_num == 0
             else: