Add test_particleclass_add_variable_in_loop

Author: VeckoTheGecko

tests/v4/test_particle.py

@@ -101,6 +101,18 @@ def test_particleclass_add_variable():
     assert p_final.variables[2].name == "varc"
 
 
+def test_particleclass_add_variable_in_loop():
+    p = ParticleClass(variables=[Variable("vara", dtype=np.float32)])
+    vars = ["sample_var", "sample_var2"]
+    for var in vars:
+        p = p.add_variable(Variable(var))
+
+    assert len(p.variables) == 3
+    assert p.variables[0].name == "vara"
+    assert p.variables[1].name == "sample_var"
+    assert p.variables[2].name == "sample_var2"
+
+
 def test_particleclass_add_variable_collision():
     p_initial = ParticleClass(variables=[Variable("vara", dtype=np.float32)])