def test_database_species_adding_and_getting():
database = Database(str(get_test_data_dir() / "supcrt98_simplified.xml"))
no_species_database = Database(
str(get_test_data_dir() / "supcrt98_no_species.xml"))
aqueous_species = database.aqueousSpecies()
gaseous_species = database.gaseousSpecies()
liquid_species = database.liquidSpecies()
mineral_species = database.mineralSpecies()
for aqueous_specie, gaseous_specie, mineral_specie, liquid_specie in zip(
aqueous_species, gaseous_species, mineral_species, liquid_species):
no_species_database.addAqueousSpecies(aqueous_specie)
no_species_database.addGaseousSpecies(gaseous_specie)
no_species_database.addLiquidSpecies(liquid_specie)
no_species_database.addMineralSpecies(mineral_specie)
for aqueous_specie in aqueous_species:
assert no_species_database.aqueousSpecies(
aqueous_specie.name()).name() == aqueous_specie.name()
for gaseous_specie in gaseous_species:
assert no_species_database.gaseousSpecies(
gaseous_specie.name()).name() == gaseous_specie.name()
for liquid_specie in liquid_species:
assert no_species_database.liquidSpecies(
liquid_specie.name()).name() == liquid_specie.name()
for mineral_specie in mineral_species:
assert no_species_database.mineralSpecies(
mineral_specie.name()).name() == mineral_specie.name()
def test_database_contains():
database = Database(str(get_test_data_dir() / "supcrt98_simplified.xml"))
aqueous_species = database.aqueousSpecies()
gaseous_species = database.gaseousSpecies()
liquid_species = database.liquidSpecies()
mineral_species = database.mineralSpecies()
assert database.containsAqueousSpecies(aqueous_species[0].name())
assert database.containsGaseousSpecies(gaseous_species[0].name())
assert database.containsLiquidSpecies(liquid_species[0].name())
assert database.containsMineralSpecies(mineral_species[0].name())