Add test for Network.get_all_species

This commit adds a test for get_all_species to ensure it outputs the correct values.
ReactionMechanismGenerator · Nov 10, 2019 · 787c7ab · 787c7ab
1 parent d618de8
commit 787c7ab
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/rmgpy/pdep/networkTest.py b/rmgpy/pdep/networkTest.py
@@ -266,6 +266,16 @@ def test_collision_matrix_memory_handling(self):
         except:
             pass
 
+    def test_get_all_species(self):
+        """
+        Ensures all species are in the get_species_list
+        """
+        species_list = self.network.get_all_species()
+        self.assertIn(self.nC4H10O, species_list)
+        self.assertIn(self.nC4H8, species_list)
+        self.assertIn(self.H2O, species_list)
+        self.assertIn(self.N2, species_list)
+
 
 ################################################################################