| @@ -283,33 +283,6 @@ class TestModelApi(TestCase): | |||||
| ['metric', 'fake_name'] | ['metric', 'fake_name'] | ||||
| ) | ) | ||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| 0 | |||||
| ) | |||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| 0.1 | |||||
| ) | |||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| True | |||||
| ) | |||||
| self.assertRaisesRegex( | self.assertRaisesRegex( | ||||
| LineageParamTypeError, | LineageParamTypeError, | ||||
| 'Element of keys must be str.', | 'Element of keys must be str.', | ||||
| @@ -319,23 +292,23 @@ class TestModelApi(TestCase): | |||||
| [1, 2, 3] | [1, 2, 3] | ||||
| ) | ) | ||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| (3, 4) | |||||
| ) | |||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| {'a': 'b'} | |||||
| ) | |||||
| @pytest.mark.level0 | |||||
| @pytest.mark.platform_arm_ascend_training | |||||
| @pytest.mark.platform_x86_gpu_training | |||||
| @pytest.mark.platform_x86_ascend_training | |||||
| @pytest.mark.platform_x86_cpu | |||||
| @pytest.mark.env_single | |||||
| def test_get_summary_lineage_exception_3(self): | |||||
| """Test the interface of get_summary_lineage with exception.""" | |||||
| for keys in [0, 0.1, True, (3, 4), {'a': 'b'}]: | |||||
| self.assertRaisesRegex( | |||||
| LineageParamTypeError, | |||||
| 'Keys must be list.', | |||||
| get_summary_lineage, | |||||
| None, | |||||
| SUMMARY_DIR, | |||||
| keys | |||||
| ) | |||||
| @pytest.mark.level0 | @pytest.mark.level0 | ||||
| @pytest.mark.platform_arm_ascend_training | @pytest.mark.platform_arm_ascend_training | ||||