[unittest] Update unit tests

Cantera · Jun 24, 2023 · fa2895f · fa2895f
1 parent cb24cea
commit fa2895f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/python/test_onedim.py b/test/python/test_onedim.py
@@ -1684,17 +1684,17 @@ def test_ion_profile(self):
         # Solution object used to compute mixture properties
         self.gas = ct.Solution('ch4_ion.yaml')
         self.gas.TPX = Tin, p, reactants
-        self.sim = ct.IonFreeFlame(self.gas, width=width)
+        self.sim = ct.FreeFlame(self.gas, width=width)
+        assert self.sim.transport_model == 'ionized-gas'
         self.sim.set_refine_criteria(ratio=4, slope=0.8, curve=1.0)
         # Ionized species may require tighter absolute tolerances
         self.sim.flame.set_steady_tolerances(Y=(1e-4, 1e-12))
-        self.sim.transport_model = 'ionized-gas'
 
         # stage one
         self.sim.solve(loglevel=0, auto=True)
 
         #stage two
-        self.sim.solve(loglevel=0, stage=2, enable_energy=True)
+        self.sim.solve(loglevel=0, stage=2)
 
         # Regression test
         self.assertNear(max(self.sim.E), 142.666221, 1e-3)
@@ -1710,10 +1710,10 @@ def test_ion_profile(self):
         # Solution object used to compute mixture properties
         self.gas = ct.Solution('ch4_ion.yaml')
         self.gas.TPX = Tburner, p, reactants
-        self.sim = ct.IonBurnerFlame(self.gas, width=width)
+        self.sim = ct.BurnerFlame(self.gas, width=width)
+        assert self.sim.transport_model == 'ionized-gas'
         self.sim.set_refine_criteria(ratio=4, slope=0.1, curve=0.1)
         self.sim.burner.mdot = self.gas.density * 0.15
-        self.sim.transport_model = 'ionized-gas'
 
         self.sim.solve(loglevel=0, stage=2, auto=True)