From be5cff059a2a5b89d7a97bc04739c4919ab8089f Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Mon, 26 Aug 2024 08:09:42 +0200
Subject: [PATCH] Test get_dataset_config_info with non-existing/gated/private
 dataset (#7124)

---
 tests/test_inspect.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/tests/test_inspect.py b/tests/test_inspect.py
index 05870e30521..63b93b3b6a6 100644
--- a/tests/test_inspect.py
+++ b/tests/test_inspect.py
@@ -1,5 +1,6 @@
 import pytest
 
+from datasets.exceptions import DatasetNotFoundError
 from datasets.inspect import (
     get_dataset_config_info,
     get_dataset_config_names,
@@ -35,11 +36,18 @@ def test_get_dataset_config_info_private(hf_token, hf_private_dataset_repo_txt_d
     "path, config_name, expected_exception",
     [
         ("paws", None, ValueError),
+        # non-existing, gated, private:
+        ("hf-internal-testing/non-existing-dataset", "default", DatasetNotFoundError),
+        ("hf-internal-testing/gated_dataset_with_data_files", "default", DatasetNotFoundError),
+        ("hf-internal-testing/private_dataset_with_data_files", "default", DatasetNotFoundError),
+        ("hf-internal-testing/gated_dataset_with_script", "default", DatasetNotFoundError),
+        ("hf-internal-testing/private_dataset_with_script", "default", DatasetNotFoundError),
     ],
 )
-def test_get_dataset_config_info_error(path, config_name, expected_exception):
+def test_get_dataset_config_info_raises(path, config_name, expected_exception):
+    kwargs = {"trust_remote_code": True} if path.endswith("_with_script") else {}
     with pytest.raises(expected_exception):
-        get_dataset_config_info(path, config_name=config_name)
+        get_dataset_config_info(path, config_name=config_name, **kwargs)
 
 
 @pytest.mark.parametrize(