Circumvent Travis CI not finding ruamel.yaml on osx

Cantera · Aug 15, 2019 · 8302778 · 8302778
1 parent e24805d
commit 8302778
Showing 1 changed file with 19 additions and 16 deletions.
diff --git a/interfaces/cython/cantera/test/test_reactor.py b/interfaces/cython/cantera/test/test_reactor.py
@@ -5,7 +5,6 @@
 
 import numpy as np
 from .utilities import unittest
-from ruamel.yaml import YAML
 
 import cantera as ct
 from . import utilities
@@ -405,24 +404,28 @@ def test_yaml(self, **kwargs):
         self.assertTrue(isinstance(reservoir.to_yaml(), str))
         self.assertTrue(isinstance(mfc.to_yaml(), str))
 
-        yaml = YAML()
-        yml = yaml.load(self.net.to_yaml())
-        self.assertTrue('ReactorNet' in yml)
-        net = yml['ReactorNet']
+        try:
+            # travis build fails on osx (ModuleNotFoundError)
+            from ruamel.yaml import YAML
 
-        self.assertTrue('ReactorBase' in net)
-        reactors = [tuple(n)[0] for n in net['ReactorBase']]
-        self.assertTrue(self.r1.name in reactors)
-        self.assertTrue(self.r2.name in reactors)
-        self.assertTrue(reservoir.name in reactors)
+            yaml = YAML()
+            yml = yaml.load(self.net.to_yaml())
+            self.assertTrue('ReactorNet' in yml)
+            net = yml['ReactorNet']
 
-        self.assertTrue('WallBase' in net)
-        walls = [tuple(n)[0] for n in net['WallBase']]
-        self.assertTrue(self.w.name in walls)
+            self.assertTrue('ReactorBase' in net)
+            reactors = [tuple(n)[0] for n in net['ReactorBase']]
+            self.assertTrue(self.r1.name in reactors)
+            self.assertTrue(self.r2.name in reactors)
+            self.assertTrue(reservoir.name in reactors)
 
-        self.assertTrue('FlowDevice' in net)
-        devices = [tuple(n)[0] for n in net['FlowDevice']]
-        self.assertTrue(mfc.name in devices)
+            self.assertTrue('WallBase' in net)
+            walls = [tuple(n)[0] for n in net['WallBase']]
+            self.assertTrue(self.w.name in walls)
+
+            self.assertTrue('FlowDevice' in net)
+            devices = [tuple(n)[0] for n in net['FlowDevice']]
+            self.assertTrue(mfc.name in devices)
 
         self.r2.name = self.r1.name
         with self.assertRaisesRegex(ct.CanteraError, 'names are not unique'):