Fix tests

yaugenst · Nov 20, 2023 · 5a7cf20 · 5a7cf20
1 parent 6a09002
commit 5a7cf20
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_fea2d.py b/tests/test_fea2d.py
@@ -16,8 +16,8 @@ def test_shape(self, fea2d_k_instance):
     def test_dofs(self, fea2d_k_instance):
         dofs = fea2d_k_instance.dofs
         assert isinstance(dofs, np.ndarray)
-        assert dofs.shape == (5, 5, 2)
-        assert np.all(dofs == np.arange(50).reshape(5, 5, 2))
+        assert dofs.shape == (50,)
+        assert np.all(dofs == np.arange(50))
 
     def test_fixdofs(self, fea2d_k_instance):
         fixdofs = fea2d_k_instance.fixdofs
@@ -42,8 +42,8 @@ def test_shape(self, fea2d_t_instance):
     def test_dofs(self, fea2d_t_instance):
         dofs = fea2d_t_instance.dofs
         assert isinstance(dofs, np.ndarray)
-        assert dofs.shape == (5, 5)
-        assert np.all(dofs == np.arange(25).reshape(5, 5))
+        assert dofs.shape == (25,)
+        assert np.all(dofs == np.arange(25))
 
     def test_fixdofs(self, fea2d_t_instance):
         fixdofs = fea2d_t_instance.fixdofs