add reference value based tests

tataratat · Jan 10, 2024 · 802b6cf · 802b6cf
1 parent 9eaef8f
commit 802b6cf
Showing 1 changed file with 153 additions and 0 deletions.
diff --git a/tests/test_utils/test_connec.py b/tests/test_utils/test_connec.py
@@ -101,6 +101,136 @@ def two_lines():
     return tl
 
 
+def quad_face_3_4():
+    return [
+        [0, 1, 4, 3],
+        [1, 2, 5, 4],
+        [3, 4, 7, 6],
+        [4, 5, 8, 7],
+        [6, 7, 10, 9],
+        [7, 8, 11, 10],
+    ]
+
+
+def quad_face_7_5():
+    return [
+        [0, 1, 8, 7],
+        [1, 2, 9, 8],
+        [2, 3, 10, 9],
+        [3, 4, 11, 10],
+        [4, 5, 12, 11],
+        [5, 6, 13, 12],
+        [7, 8, 15, 14],
+        [8, 9, 16, 15],
+        [9, 10, 17, 16],
+        [10, 11, 18, 17],
+        [11, 12, 19, 18],
+        [12, 13, 20, 19],
+        [14, 15, 22, 21],
+        [15, 16, 23, 22],
+        [16, 17, 24, 23],
+        [17, 18, 25, 24],
+        [18, 19, 26, 25],
+        [19, 20, 27, 26],
+        [21, 22, 29, 28],
+        [22, 23, 30, 29],
+        [23, 24, 31, 30],
+        [24, 25, 32, 31],
+        [25, 26, 33, 32],
+        [26, 27, 34, 33],
+    ]
+
+
+def hexa_volume_2_3_4():
+    return [
+        [0, 1, 3, 2, 6, 7, 9, 8],
+        [2, 3, 5, 4, 8, 9, 11, 10],
+        [6, 7, 9, 8, 12, 13, 15, 14],
+        [8, 9, 11, 10, 14, 15, 17, 16],
+        [12, 13, 15, 14, 18, 19, 21, 20],
+        [14, 15, 17, 16, 20, 21, 23, 22],
+    ]
+
+
+def hexa_volume_5_7_3():
+    return [
+        [0, 1, 6, 5, 35, 36, 41, 40],
+        [1, 2, 7, 6, 36, 37, 42, 41],
+        [2, 3, 8, 7, 37, 38, 43, 42],
+        [3, 4, 9, 8, 38, 39, 44, 43],
+        [5, 6, 11, 10, 40, 41, 46, 45],
+        [6, 7, 12, 11, 41, 42, 47, 46],
+        [7, 8, 13, 12, 42, 43, 48, 47],
+        [8, 9, 14, 13, 43, 44, 49, 48],
+        [10, 11, 16, 15, 45, 46, 51, 50],
+        [11, 12, 17, 16, 46, 47, 52, 51],
+        [12, 13, 18, 17, 47, 48, 53, 52],
+        [13, 14, 19, 18, 48, 49, 54, 53],
+        [15, 16, 21, 20, 50, 51, 56, 55],
+        [16, 17, 22, 21, 51, 52, 57, 56],
+        [17, 18, 23, 22, 52, 53, 58, 57],
+        [18, 19, 24, 23, 53, 54, 59, 58],
+        [20, 21, 26, 25, 55, 56, 61, 60],
+        [21, 22, 27, 26, 56, 57, 62, 61],
+        [22, 23, 28, 27, 57, 58, 63, 62],
+        [23, 24, 29, 28, 58, 59, 64, 63],
+        [25, 26, 31, 30, 60, 61, 66, 65],
+        [26, 27, 32, 31, 61, 62, 67, 66],
+        [27, 28, 33, 32, 62, 63, 68, 67],
+        [28, 29, 34, 33, 63, 64, 69, 68],
+        [35, 36, 41, 40, 70, 71, 76, 75],
+        [36, 37, 42, 41, 71, 72, 77, 76],
+        [37, 38, 43, 42, 72, 73, 78, 77],
+        [38, 39, 44, 43, 73, 74, 79, 78],
+        [40, 41, 46, 45, 75, 76, 81, 80],
+        [41, 42, 47, 46, 76, 77, 82, 81],
+        [42, 43, 48, 47, 77, 78, 83, 82],
+        [43, 44, 49, 48, 78, 79, 84, 83],
+        [45, 46, 51, 50, 80, 81, 86, 85],
+        [46, 47, 52, 51, 81, 82, 87, 86],
+        [47, 48, 53, 52, 82, 83, 88, 87],
+        [48, 49, 54, 53, 83, 84, 89, 88],
+        [50, 51, 56, 55, 85, 86, 91, 90],
+        [51, 52, 57, 56, 86, 87, 92, 91],
+        [52, 53, 58, 57, 87, 88, 93, 92],
+        [53, 54, 59, 58, 88, 89, 94, 93],
+        [55, 56, 61, 60, 90, 91, 96, 95],
+        [56, 57, 62, 61, 91, 92, 97, 96],
+        [57, 58, 63, 62, 92, 93, 98, 97],
+        [58, 59, 64, 63, 93, 94, 99, 98],
+        [60, 61, 66, 65, 95, 96, 101, 100],
+        [61, 62, 67, 66, 96, 97, 102, 101],
+        [62, 63, 68, 67, 97, 98, 103, 102],
+        [63, 64, 69, 68, 98, 99, 104, 103],
+    ]
+
+
+def hexa_volume_4_8_2():
+    return [
+        [0, 1, 5, 4, 32, 33, 37, 36],
+        [1, 2, 6, 5, 33, 34, 38, 37],
+        [2, 3, 7, 6, 34, 35, 39, 38],
+        [4, 5, 9, 8, 36, 37, 41, 40],
+        [5, 6, 10, 9, 37, 38, 42, 41],
+        [6, 7, 11, 10, 38, 39, 43, 42],
+        [8, 9, 13, 12, 40, 41, 45, 44],
+        [9, 10, 14, 13, 41, 42, 46, 45],
+        [10, 11, 15, 14, 42, 43, 47, 46],
+        [12, 13, 17, 16, 44, 45, 49, 48],
+        [13, 14, 18, 17, 45, 46, 50, 49],
+        [14, 15, 19, 18, 46, 47, 51, 50],
+        [16, 17, 21, 20, 48, 49, 53, 52],
+        [17, 18, 22, 21, 49, 50, 54, 53],
+        [18, 19, 23, 22, 50, 51, 55, 54],
+        [20, 21, 25, 24, 52, 53, 57, 56],
+        [21, 22, 26, 25, 53, 54, 58, 57],
+        [22, 23, 27, 26, 54, 55, 59, 58],
+        [24, 25, 29, 28, 56, 57, 61, 60],
+        [25, 26, 30, 29, 57, 58, 62, 61],
+        [26, 27, 31, 30, 58, 59, 63, 62],
+    ]
+
+
 def test_sequentialize_directed_edges():
     # polygon
     # as default, it starts at minimum
@@ -239,3 +369,26 @@ def test_faces_to_edges():
     expected = np.vstack(expected).reshape(-1, 2)
 
     assert (gus.utils.connec.faces_to_edges(query) == expected).all()
+
+
+def test_make_quad_faces():
+    """
+    checks against reference value. Feel free to extend!
+    """
+    assert (gus.utils.connec.make_quad_faces([3, 4]) == quad_face_3_4()).all()
+    assert (gus.utils.connec.make_quad_faces([7, 5]) == quad_face_7_5()).all()
+
+
+def test_make_hexa_volumes():
+    """
+    checks against reference value. Feel free to extend!
+    """
+    assert (
+        gus.utils.connec.make_hexa_volumes([4, 8, 2]) == hexa_volume_4_8_2()
+    ).all()
+    assert (
+        gus.utils.connec.make_hexa_volumes([2, 3, 4]) == hexa_volume_2_3_4()
+    ).all()
+    assert (
+        gus.utils.connec.make_hexa_volumes([5, 7, 3]) == hexa_volume_5_7_3()
+    ).all()