Apply suggestions from code review

Co-authored-by: James Dark <[email protected]>
festim-dev · Oct 28, 2024 · fb46ab6 · fb46ab6
1 parent d31bcb8
commit fb46ab6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/unit/test_exports/test_derived_quantities/test_derived_quantities.py b/test/unit/test_exports/test_derived_quantities/test_derived_quantities.py
@@ -57,8 +57,7 @@ def test_simple(self):
         """
         my_derv_quant = DerivedQuantities([self.surface_flux_1])
 
-        for quantity in my_derv_quant:
-            quantity.function = f.Function(self.V)
+        my_derv_quant[0].function = f.Function(self.V)
 
         header = my_derv_quant.make_header()