def test_equilibrium_problem_add_by_chemical_state(partition_with_inert_gaseous_phase, chemical_system):
state = ChemicalState(chemical_system)
state.setSpeciesAmount("CO2(g)", 10.0)
problem = EquilibriumProblem(partition_with_inert_gaseous_phase)
problem.add(state)
for element in problem.elementAmounts():
assert element == 0.0
assert problem.partition().numInertSpecies() == 2
def test_equilibrium_problem_add(partition_with_inert_gaseous_phase):
problem = EquilibriumProblem(partition_with_inert_gaseous_phase)
problem.add("CO2", 10.0, 'mol')
assert sum(problem.elementAmounts()) == 30.0
assert problem.partition().numInertSpecies() == 2
