示例#1
0
 def test_get_dataset_path_unknown_framework(self) -> None:
     """Test getting dataset path failure."""
     framework = "unknown"
     domain = "image_recognition"
     with self.assertRaises(Exception):
         get_dataset_path(framework, domain)
示例#2
0
 def test_get_dataset_path_unknown_domain(self) -> None:
     """Test getting dataset path failure."""
     framework = "tensorflow"
     domain = "domain"
     with self.assertRaises(Exception):
         get_dataset_path(framework, domain)
示例#3
0
 def test_get_dataset_path(self) -> None:
     """Test getting dataset path."""
     framework = "tensorflow"
     domain = "image_recognition"
     result = get_dataset_path(framework, domain)
     self.assertEqual(result, "examples/test/dataset/imagenet")