def test_gpu_distributed_equipped(self): clear_capacity() os.environ[sampcd_processor.ENV_KEY_TEST_CAPACITY] = 'gpu,distributed' get_test_capacity() self.assertTrue(is_required_match('cpu')) self.assertTrue(is_required_match('gpu')) self.assertTrue(is_required_match('distributed')) self.assertFalse(is_required_match('xpu')) self.assertIsNone(is_required_match('skiptest'))
def test_alldefault(self): clear_capacity() get_test_capacity() self.assertTrue(is_required_match('')) self.assertTrue(is_required_match(None)) self.assertTrue(is_required_match('cpu')) self.assertFalse(is_required_match('gpu')) self.assertIsNone(is_required_match('skiptest')) self.assertIsNone(is_required_match('skip')) self.assertIsNone(is_required_match('cpu,skiptest'))
def test_2_samplecodes_has_skipped(self): comments = """ placeholder Examples: .. code-block:: python # required: skiptest print(1/0) .. code-block:: python print(1+1) .. code-block:: python # required: gpu print(1//1) .. code-block:: python # required: xpu print(1//1) .. code-block:: python # required: distributed print(1//1) .. code-block:: python # required: gpu print(1//1) """ funcname = 'one_plus_one' clear_summary_info() clear_capacity() get_test_capacity() sample_code_filenames = sampcd_extract_to_file(comments, funcname) self.assertCountEqual([ os.path.join(sampcd_processor.SAMPLECODE_TEMPDIR, funcname + '_example_2.py') ], sample_code_filenames) self.assertCountEqual(sampcd_processor.SUMMARY_INFO['skiptest'], [funcname + '-1']) self.assertCountEqual(sampcd_processor.SUMMARY_INFO['gpu'], [funcname + '-3', funcname + '-6']) self.assertCountEqual(sampcd_processor.SUMMARY_INFO['xpu'], [funcname + '-4']) self.assertCountEqual(sampcd_processor.SUMMARY_INFO['distributed'], [funcname + '-5'])
def tearDown(self): shutil.rmtree(sampcd_processor.SAMPLECODE_TEMPDIR) clear_capacity() get_test_capacity()
def setUp(self): if not os.path.exists(sampcd_processor.SAMPLECODE_TEMPDIR): os.mkdir(sampcd_processor.SAMPLECODE_TEMPDIR) clear_capacity() os.environ[sampcd_processor.ENV_KEY_TEST_CAPACITY] = 'gpu,distributed' get_test_capacity()
def tearDown(self): clear_capacity() get_test_capacity()
def test_EnvVar_gpu_and_distributed(self): clear_capacity() os.environ[sampcd_processor.ENV_KEY_TEST_CAPACITY] = 'gpu,distributed' get_test_capacity() self.assertCountEqual(['cpu', 'gpu', 'distributed'], sampcd_processor.SAMPLE_CODE_TEST_CAPACITY)
def test_NoEnvVar_RUN_ON_DEVICE_gpu(self): clear_capacity() sampcd_processor.RUN_ON_DEVICE = 'gpu' get_test_capacity() self.assertCountEqual(['cpu', 'gpu'], sampcd_processor.SAMPLE_CODE_TEST_CAPACITY)
def test_NoEnvVar(self): clear_capacity() get_test_capacity() self.assertCountEqual([ 'cpu', ], sampcd_processor.SAMPLE_CODE_TEST_CAPACITY)
def setUp(self): clear_capacity() get_test_capacity()