Re-add helper functions to assert equality of FSArray instances

Author: sebastinas

curtsies/formatstringarray.py

@@ -297,6 +297,37 @@ def simple_format(x):
     return "\n".join(actualize(l) for l in x)
 
 
+def assertFSArraysEqual(a, b):
+    # type: (FSArray, FSArray) -> None
+    assert isinstance(a, FSArray)
+    assert isinstance(b, FSArray)
+    assert (
+        a.width == b.width and a.hight == b.hight
+    ), f"fsarray dimensions do not match: {a.shape} {b.shape}"
+    for i, (a_row, b_row) in enumerate(zip(a, b)):
+        assert a_row == b_row, "FSArrays differ first on line {}:\n{}".format(
+            i, FSArray.diff(a, b)
+        )
+
+
+def assertFSArraysEqualIgnoringFormatting(a, b):
+    # type: (FSArray, FSArray) -> None
+    """Also accepts arrays of strings"""
+    assert len(a) == len(b), "fsarray heights do not match: %s %s \n%s \n%s" % (
+        len(a),
+        len(b),
+        simple_format(a),
+        simple_format(b),
+    )
+    for i, (a_row, b_row) in enumerate(zip(a, b)):
+        a_row = a_row.s if isinstance(a_row, FmtStr) else a_row
+        b_row = b_row.s if isinstance(b_row, FmtStr) else b_row
+        assert a_row == b_row, "FSArrays differ first on line %s:\n%s" % (
+            i,
+            FSArray.diff(a, b, ignore_formatting=True),
+        )
+
+
 if __name__ == "__main__":
     a = FSArray(3, 14, bg="blue")
     a[0:2, 5:11] = cast(