Skip to content

Commit

Permalink
Tests: Scheduler species_has_x() tests for species with yml_path attr
Browse files Browse the repository at this point in the history
  • Loading branch information
alongd committed Oct 2, 2023
1 parent 659bf33 commit e8e3fe4
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion arc/scheduler_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from arc.job.factory import job_factory
from arc.level import Level
from arc.plotter import save_conformers_file
from arc.scheduler import Scheduler, species_has_freq, species_has_geo, species_has_sp
from arc.scheduler import Scheduler, species_has_freq, species_has_geo, species_has_sp, species_has_sp_and_freq
from arc.imports import settings
from arc.reaction import ARCReaction
from arc.species.species import ARCSpecies
Expand Down Expand Up @@ -667,6 +667,12 @@ def test_species_has_geo_sp_freq(self):
self.assertTrue(species_has_property((species_output_dict)))
species_output_dict = {'paths': {property_: True, 'composite': True}}
self.assertTrue(species_has_property((species_output_dict)))
yml_path=os.path.join(ARC_PATH, 'arc', 'testing', 'yml_testing', 'N4H6.yml')
species_output_dict = {'paths': {'geo': False, 'sp': False, 'freq': False, 'composite': False}}
self.assertTrue(species_has_freq(species_output_dict=species_output_dict, yml_path=yml_path))
self.assertTrue(species_has_geo(species_output_dict=species_output_dict, yml_path=yml_path))
self.assertTrue(species_has_sp(species_output_dict=species_output_dict, yml_path=yml_path))
self.assertTrue(species_has_sp_and_freq(species_output_dict=species_output_dict, yml_path=yml_path))

def test_add_label_to_unique_species_labels(self):
"""Test the add_label_to_unique_species_labels() method."""
Expand Down

0 comments on commit e8e3fe4

Please sign in to comment.