diff options
Diffstat (limited to 'test-chill/unit-tests/test_chill.py')
| -rw-r--r-- | test-chill/unit-tests/test_chill.py | 215 |
1 files changed, 215 insertions, 0 deletions
diff --git a/test-chill/unit-tests/test_chill.py b/test-chill/unit-tests/test_chill.py new file mode 100644 index 0000000..8aaebfe --- /dev/null +++ b/test-chill/unit-tests/test_chill.py @@ -0,0 +1,215 @@ +import logging +import os +import unittest + +import testchill.chill +import testchill.gcov +import testchill.test +import testchill.util + + +_runbuild=True + +def runtest(tclist): + if _runbuild: + for tc in tclist: + tc.setUp() + tc.setresult(tc.run()) + tc.tearDown() + +def runchilltest(tclist, runvalidate=False, runstdout=False): + for tc in tclist: + tc.setUp() + tc.compile_src(tc) + tc.run_script(tc) + tc.compile_gensrc(tc) + if runvalidate: + tc.check_run_script_validate(tc) + if runstdout: + tc.check_run_script_stdout(tc) + tc.tearDown() + +class TestChillTestCases(unittest.TestCase): + def config(self, **kwargs): + cargs = { + 'omega_dir': self.omega_dev_dir, + 'chill_dir': self.chill_dev_dir, + 'bin_dir': self.bin_dir, + 'build_cuda': False, + 'script_lang': None, + 'version': 'dev' + } + cargs.update(kwargs) + return testchill.chill.ChillConfig(**cargs) + + def config_rel(self, **kwargs): + kwargs['version'] = 'release' + kwargs['omega_dir'] = self.omega_rel_dir + kwargs['chill_dir'] = self.chill_rel_dir + return self.config(**kwargs) + + def setUp(self): + self.chill_dev_dir = os.getenv('CHILL_DEV_SRC') + self.chill_rel_dir = os.getenv('CHILL_RELEASE_SRC') + self.omega_dev_dir = os.getenv('OMEGA_DEV_SRC') + self.omega_rel_dir = os.getenv('OMEGA_RELEASE_SRC') + self.bin_dir = os.getenv('STAGING_DIR_BIN') + self.wd = os.getenv('STAGING_DIR_WD') + self.build_options = {'coverage':False} + + testchill.util.shell('cp', [os.path.join(self.chill_dev_dir, 'examples/cuda-chill/cudaize.lua'), self.wd]) + testchill.util.shell('cp', [os.path.join(self.chill_dev_dir, 'examples/cuda-chill/cudaize.py'), self.wd]) + + self.config_test_func = { + 0: lambda conf: conf.default_script_lang(), + 1: lambda conf: conf.name(), + 2: lambda conf: conf.make_depend_target(), + 3: lambda conf: conf.make_target(), + 4: lambda conf: conf.make_args() + } + self.config_test_data = [ + ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, None, 'dev'), ('script', 'chill', 'depend-chill', 'chill')), + ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, 'lua', 'dev'), ('script', 'chill-lua', 'depend-chill', 'chill')), + ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, 'python', 'dev'), ('script', 'chill-python', 'depend-chill', 'chill')), + ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, True, None, 'dev'), ('lua', 'cuda-chill', 'depend-cuda-chill', 'cuda-chill')), + ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, True, 'python', 'dev'), ('lua', 'cuda-chill-python', 'depend-cuda-chill', 'cuda-chill')), + ((self.omega_rel_dir, self.chill_rel_dir, self.bin_dir, False, None, 'release'), ('script', 'chill-release', 'depend', 'chill')), + ((self.omega_rel_dir, self.chill_rel_dir, self.bin_dir, True, None, 'release'), ('lua', 'cuda-chill-release', 'depend-cuda-chill', 'cuda-chill')) + ] + + def tearDown(self): + pass + + def _run_ChillConfig_test(self, n): + for args, expected in self.config_test_data: + val = self.config_test_func[n](testchill.chill.ChillConfig(*args)) + exp = expected[n] + self.assertEqual(val, exp) + + def test_ChillConfig_default_script_lang(self): + self._run_ChillConfig_test(0) + + def test_ChillConfig_name(self): + self._run_ChillConfig_test(1) + + def test_ChillConfig_make_depend_target(self): + self._run_ChillConfig_test(2) + + def test_ChillConfig_make_target(self): + self._run_ChillConfig_test(3) + + #def test_ChillConfig_make_args(self): + # self._run_ChillConfig_test(4) + + def test_chill_dev(self): + tc = testchill.chill.BuildChillTestCase(self.config(), self.build_options) + self.assertEqual(tc.config.name(), 'chill') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-chill') + self.assertEqual(tc.config.make_target(), 'chill') + self.assertEqual(tc.name, 'chill') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_chill_dev_lua(self): + tc = testchill.chill.BuildChillTestCase(self.config(script_lang='lua'), self.build_options) + self.assertEqual(tc.config.name(), 'chill-lua') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-chill') + self.assertEqual(tc.config.make_target(), 'chill') + self.assertEqual(tc.config.make_args(), 'SCRIPT_LANG=lua') + self.assertEqual(tc.name, 'chill-lua') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_chill_dev_python(self): + tc = testchill.chill.BuildChillTestCase(self.config(script_lang='python'), self.build_options) + self.assertEqual(tc.config.name(), 'chill-python') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-chill') + self.assertEqual(tc.config.make_target(), 'chill') + self.assertEqual(tc.config.make_args(), 'SCRIPT_LANG=python') + self.assertEqual(tc.name, 'chill-python') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_cudachill_dev(self): + tc = testchill.chill.BuildChillTestCase(self.config(build_cuda=True), self.build_options) + self.assertEqual(tc.config.name(), 'cuda-chill') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill') + self.assertEqual(tc.config.make_target(), 'cuda-chill') + self.assertEqual(tc.name, 'cuda-chill') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_cudachill_dev(self): + tc = testchill.chill.BuildChillTestCase(self.config(build_cuda=True, script_lang='python'), self.build_options) + self.assertEqual(tc.config.name(), 'cuda-chill-python') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill') + self.assertEqual(tc.config.make_target(), 'cuda-chill') + self.assertEqual(tc.name, 'cuda-chill-python') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_chill_release(self): + tc = testchill.chill.BuildChillTestCase(self.config_rel(), self.build_options) + self.assertEqual(tc.config.name(), 'chill-release') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_rel_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend') + self.assertEqual(tc.config.make_target(), 'chill') + self.assertEqual(tc.name, 'chill-release') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_cudachill_release(self): + tc = testchill.chill.BuildChillTestCase(self.config_rel(build_cuda=True), self.build_options) + self.assertEqual(tc.config.name(), 'cuda-chill-release') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_rel_dir) + self.assertEqual(tc.config.env()['CUDACHILL'], 'true') + self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill') + self.assertEqual(tc.config.make_target(), 'cuda-chill') + self.assertEqual(tc.name, 'cuda-chill-release') + logging.info('Building ' + tc.name) + runtest([tc]) + + def test_run_chill(self): + config = self.config() + btc = testchill.chill.BuildChillTestCase(config, self.build_options) + runtest([btc]) + tc = testchill.chill.RunChillTestCase(config, 'test-cases/chill/test_scale.script', 'test-cases/chill/mm.c', wd=self.wd) + self.assertEqual(tc.chill_src, 'mm.c') + self.assertEqual(tc.chill_script, 'test_scale.script') + self.assertEqual(tc.chill_src_path, os.path.join(os.getcwd(), 'test-cases/chill/mm.c')) + self.assertEqual(tc.chill_script_path, os.path.join(os.getcwd(), 'test-cases/chill/test_scale.script')) + self.assertEqual(tc.chill_gensrc, 'rose_mm.c') + self.assertEqual(tc.name, 'chill:test_scale.script') + runchilltest([tc]) + + def test_run_cudachill(self): + config = self.config(build_cuda=True) + btc = testchill.chill.BuildChillTestCase(config, self.build_options) + runtest([btc]) + tc = testchill.chill.RunChillTestCase(config, 'test-cases/examples/cuda-chill/mm.lua', 'test-cases/examples/cuda-chill/mm.c', wd=self.wd) + self.assertEqual(tc.chill_src, 'mm.c') + self.assertEqual(tc.chill_script, 'mm.lua') + self.assertEqual(tc.chill_src_path, os.path.join(os.getcwd(), 'test-cases/examples/cuda-chill/mm.c')) + self.assertEqual(tc.chill_script_path, os.path.join(os.getcwd(), 'test-cases/examples/cuda-chill/mm.lua')) + self.assertEqual(tc.chill_gensrc, 'rose_mm.cu') + self.assertEqual(tc.name, 'cuda-chill:mm.lua') + runchilltest([tc]) + + def test_chill_coverage(self): + tc = testchill.chill.BuildChillTestCase(self.config(), options={'coverage':True}, coverage_set=testchill.gcov.GcovSet()) + self.assertEqual(tc.config.name(), 'chill') + self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir) + self.assertEqual(tc.config.make_depend_target(), 'depend-chill') + self.assertEqual(tc.config.make_target(), 'chill') + self.assertEqual(tc.name, 'chill') + self.assertTrue(tc.options['coverage']) + logging.info('Building ' + tc.name) + if _runbuild: + runtest([tc]) + self.assertTrue(os.path.exists(os.path.join(self.chill_dev_dir, 'ir_rose.gcno'))) + |