Merge pull request #145 from granrothge/master

Fixes yaml loader issue with yaml 6.0
neutrons · Aug 19, 2022 · ca045fb · ca045fb
2 parents 508ab31 + 75faa96
commit ca045fb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/multiphonon/ui/__init__.py b/multiphonon/ui/__init__.py
@@ -31,7 +31,7 @@ def to_yaml(self, path):
         return
     def from_yaml(self, path):
         with open(path) as stream:
-            d = yaml.load(stream)
+            d = yaml.load(stream,Loader=yaml.FullLoader)
         for k, v in d.items():
             setattr(self, k,v)
             continue

diff --git a/tests/ui/getdos_TestCase.py b/tests/ui/getdos_TestCase.py
@@ -16,6 +16,14 @@
 import unittest
 class TestCase(unittest.TestCase):
 
+    def test_contextload(self):
+        from multiphonon.ui import Context
+        cntxt =  Context()
+        cntxt.to_yaml('context.yaml')
+        cntxt2 = Context()
+        cntxt2.from_yaml('context.yaml')
+        return
+
     def test1(self):
         "multiphonon.ui.getdos"
         from multiphonon.ui import getdos