public release 2.2.0 - see README.md and CHANGES.md for details

tests/calculators/phagen/test_translator.py

@@ -0,0 +1,79 @@
+"""
+@package tests.calculators.phagen.test_translator
+unit tests for pmsco.calculators.phagen.translator
+
+the purpose of these tests is to check whether the code runs as expected in a particular environment.
+
+to run the tests, change to the directory which contains the tests directory, and execute =nosetests=.
+
+@pre nose must be installed (python-nose package on Debian).
+
+@author Matthias Muntwiler, matthias.muntwiler@psi.ch
+
+@copyright (c) 2015-19 by Paul Scherrer Institut @n
+Licensed under the Apache License, Version 2.0 (the "License"); @n
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+  http://www.apache.org/licenses/LICENSE-2.0
+"""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+import unittest
+from pmsco.calculators.phagen import translator
+
+
+class TestModule(unittest.TestCase):
+    def setUp(self):
+        # before each test method
+        pass
+
+    def tearDown(self):
+        # after each test method
+        pass
+
+    @classmethod
+    def setup_class(cls):
+        # before any methods in this class
+        pass
+
+    @classmethod
+    def teardown_class(cls):
+        # teardown_class() after any methods in this class
+        pass
+
+    def test_state_to_edge(self):
+        self.assertEqual(translator.state_to_edge('1s'), 'k')
+        self.assertEqual(translator.state_to_edge('2s'), 'l1')
+        self.assertEqual(translator.state_to_edge('3s'), 'm1')
+        self.assertEqual(translator.state_to_edge('4s'), 'n1')
+        self.assertEqual(translator.state_to_edge('5s'), 'o1')
+
+        self.assertEqual(translator.state_to_edge('2p'), 'l2')
+        self.assertEqual(translator.state_to_edge('3p'), 'm2')
+        self.assertEqual(translator.state_to_edge('4p'), 'n2')
+        self.assertEqual(translator.state_to_edge('5p'), 'o2')
+
+        self.assertEqual(translator.state_to_edge('2p1/2'), 'l2')
+        self.assertEqual(translator.state_to_edge('3p1/2'), 'm2')
+        self.assertEqual(translator.state_to_edge('4p1/2'), 'n2')
+        self.assertEqual(translator.state_to_edge('5p1/2'), 'o2')
+
+        self.assertEqual(translator.state_to_edge('2p3/2'), 'l3')
+        self.assertEqual(translator.state_to_edge('3p3/2'), 'm3')
+        self.assertEqual(translator.state_to_edge('4p3/2'), 'n3')
+        self.assertEqual(translator.state_to_edge('5p3/2'), 'o3')
+
+        self.assertEqual(translator.state_to_edge('3d'), 'm4')
+        self.assertEqual(translator.state_to_edge('4d'), 'n4')
+        self.assertEqual(translator.state_to_edge('5d'), 'o4')
+
+        self.assertEqual(translator.state_to_edge('3d3/2'), 'm4')
+        self.assertEqual(translator.state_to_edge('4d3/2'), 'n4')
+        self.assertEqual(translator.state_to_edge('5d3/2'), 'o4')
+
+        self.assertEqual(translator.state_to_edge('3d5/2'), 'm5')
+        self.assertEqual(translator.state_to_edge('4d5/2'), 'n5')
+        self.assertEqual(translator.state_to_edge('5d5/2'), 'o5')