Mentions légales du service

Skip to content
Snippets Groups Projects
Commit a39b7fff authored by hhakim's avatar hhakim
Browse files

Add a unit test for ConstraintName.str2name_int, name_int2str.

parent 251d0961
Branches
No related tags found
No related merge requests found
Pipeline #889299 passed
import unittest
from pyfaust.factparams import ConstraintName
class TestFactParams(unittest.TestCase):
def __init__(self, methodName='runTest', dev='cpu', dtype='double'):
super(TestFactParams, self).__init__(methodName)
def test_int2str_str2int(self):
print("ConstraintName.name_int2str & name_str2int")
max_int_name = 0
for attr in ConstraintName.__dict__:
if isinstance(ConstraintName.__dict__[attr], int):
str_name = attr.lower().replace('_', '').replace('blkdiag',
'blockdiag')
self.assertEqual(ConstraintName.name_int2str(ConstraintName.__dict__[attr]),
str_name)
self.assertEqual(ConstraintName.__dict__[attr],
ConstraintName.str2name_int(str_name))
if ConstraintName.__dict__[attr] > max_int_name:
max_int_name = ConstraintName.__dict__[attr]
err_msg = "Invalid argument to designate a ConstraintName."
self.assertRaisesRegex(ValueError, err_msg,
ConstraintName.str2name_int, "notaconstraint")
self.assertRaisesRegex(ValueError, err_msg,
ConstraintName.name_int2str, max_int_name+1)
import unittest import unittest
from pyfaust.tests.TestFaust import TestFaust from pyfaust.tests.TestFaust import TestFaust
from pyfaust.tests.TestPoly import TestPoly from pyfaust.tests.TestPoly import TestPoly
from pyfaust.tests.TestFactParams import TestFactParams
def run_tests(dev, dtype): def run_tests(dev, dtype):
...@@ -10,7 +11,7 @@ def run_tests(dev, dtype): ...@@ -10,7 +11,7 @@ def run_tests(dev, dtype):
""" """
runner = unittest.TextTestRunner() runner = unittest.TextTestRunner()
suite = unittest.TestSuite() suite = unittest.TestSuite()
for class_name in ['TestFaust', 'TestPoly']: for class_name in ['TestFaust', 'TestPoly', 'TestFactParams']:
testloader = unittest.TestLoader() testloader = unittest.TestLoader()
test_names = eval("testloader.getTestCaseNames("+class_name+")") test_names = eval("testloader.getTestCaseNames("+class_name+")")
for meth_name in test_names: for meth_name in test_names:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment