Add more tests

src/felupe/assembly/_integral.py

@@ -283,22 +283,15 @@ def assemble(self, values=None, parallel=False, block=True, out=None):
         for val, form in zip(values, self.forms):
             res.append(form.assemble(val, parallel=parallel, out=out))
 
-        if block and self.mode == 2:
+        if block and (self.mode == 2 or self.mode == 3):
             K = np.zeros((self.nv, self.nu), dtype=object)
             for a, (i, j) in enumerate(zip(self.i, self.j)):
                 K[i, j] = res[a]
-                if i != j:
+                if self.mode == 2 and i != j:
                     K[j, i] = res[a].T
 
             res = bmat(K).tocsr()
 
-        if block and self.mode == 3:
-            K = np.zeros((self.nv, self.nu), dtype=object)
-            for a, (i, j) in enumerate(zip(self.i, self.j)):
-                K[i, j] = res[a]
-
-            res = bmat(K).tocsr()
-
         if block and self.mode == 1:
             res = vstack(res).tocsr()
 

tests/test_form.py

@@ -252,6 +252,20 @@ def test_mixed():
 
         assert b.shape == (z, 1)
 
+    L = fem.IntegralForm([f[0], None, f[2]], v, r.dV)
+    x = L.integrate()
+    b = L.assemble(x)
+
+    assert b.shape == (z, 1)
+
+    a = fem.IntegralForm(
+        [A[0], A[1], A[2], A[1], A[3], A[4], A[2], A[4], A[5]], v, r.dV, v
+    )
+    y = a.integrate()
+    K = a.assemble(y)
+
+    assert K.shape == (z, z)
+
     r, v, f, A = pre_axi_mixed()
 
     for parallel in [False, True]: