80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
"""
|
|
@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')
|