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)
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)
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")