Move remove_from_cache to tests

Author: alejoe91

src/probeinterface/library.py

@@ -102,38 +102,6 @@ def get_from_cache(manufacturer: str, probe_name: str, tag: Optional[str] = None
         return probe
 
 
-def remove_from_cache(manufacturer: str, probe_name: str, tag: Optional[str] = None) -> Optional["Probe"]:
-    """
-    Remove Probe from local cache
-
-    Parameters
-    ----------
-    manufacturer : "cambridgeneurotech" | "neuronexus" | "plexon" | "imec" | "sinaps"
-        The probe manufacturer
-    probe_name : str (see probeinterface_libary for options)
-        The probe name
-    tag : str | None, default: None
-        Optional tag for the probe
-
-    Returns
-    -------
-    probe : Probe object, or None if no probeinterface JSON file is found
-
-    """
-    cache_folder = get_cache_folder()
-    if tag is not None:
-        cache_folder_tag = cache_folder / tag
-        if not cache_folder_tag.is_dir():
-            return None
-        cache_folder = cache_folder_tag
-    else:
-        cache_folder_tag = cache_folder / "main"
-
-    local_file = cache_folder_tag / manufacturer / (probe_name + ".json")
-    if local_file.is_file():
-        os.remove(local_file)
-
-
 def get_probe(
     manufacturer: str,
     probe_name: str,

tests/test_library.py

@@ -1,8 +1,8 @@
+import os
 from probeinterface import Probe
 from probeinterface.library import (
     download_probeinterface_file,
     get_from_cache,
-    remove_from_cache,
     get_probe,
     get_tags_in_library,
     list_manufacturers_in_library,
@@ -15,6 +15,33 @@
 probe_name = "A1x32-Poly3-10mm-50-177"
 
 
+def _remove_from_cache(manufacturer: str, probe_name: str, tag = None) -> None:
+    """
+    Remove Probe from local cache
+
+    Parameters
+    ----------
+    manufacturer : "cambridgeneurotech" | "neuronexus" | "plexon" | "imec" | "sinaps"
+        The probe manufacturer
+    probe_name : str (see probeinterface_libary for options)
+        The probe name
+    tag : str | None, default: None
+        Optional tag for the probe
+    """
+    cache_folder = get_cache_folder()
+    if tag is not None:
+        cache_folder_tag = cache_folder / tag
+        if not cache_folder_tag.is_dir():
+            return None
+        cache_folder = cache_folder_tag
+    else:
+        cache_folder_tag = cache_folder / "main"
+
+    local_file = cache_folder_tag / manufacturer / (probe_name + ".json")
+    if local_file.is_file():
+        os.remove(local_file)
+
+
 def test_download_probeinterface_file():
     download_probeinterface_file(manufacturer, probe_name, tag=None)
 
@@ -29,7 +56,7 @@ def test_get_from_cache():
     assert probe is None  # because we did not download with this tag
     download_probeinterface_file(manufacturer, probe_name, tag=tag)
     probe = get_from_cache(manufacturer, probe_name, tag=tag)
-    remove_from_cache(manufacturer, probe_name, tag=tag)
+    _remove_from_cache(manufacturer, probe_name, tag=tag)
     assert isinstance(probe, Probe)
 
     probe = get_from_cache("yep", "yop")