test private API using test class

src/databricks/labs/ucx/source_code/notebooks/loaders.py

@@ -55,14 +55,14 @@ def load_dependency(self, path_lookup: PathLookup, dependency: Dependency) -> So
         except NotFound:
             logger.warning(f"Could not read notebook from workspace: {absolute_path}")
             return None
-        language = self.detect_language(absolute_path, content)
+        language = self._detect_language(absolute_path, content)
         if not language:
             logger.warning(f"Could not detect language for {absolute_path}")
             return None
         return Notebook.parse(absolute_path, content, language)
 
     @staticmethod
-    def detect_language(path: Path, content: str):
+    def _detect_language(path: Path, content: str):
         language = SUPPORTED_EXTENSION_LANGUAGES.get(path.suffix, None)
         if language:
             return language

tests/unit/source_code/linters/test_python_ast.py

@@ -1,4 +1,3 @@
-
 import pytest
 from astroid import Attribute, Call, Const, Expr  # type: ignore
 

tests/unit/source_code/notebooks/test_loader.py

@@ -5,8 +5,15 @@
 
 
 def test_detects_language():
-    assert NotebookLoader.detect_language(Path("hi.py"), "stuff") == Language.PYTHON
-    assert NotebookLoader.detect_language(Path("hi.sql"), "stuff") == Language.SQL
-    assert NotebookLoader.detect_language(Path("hi"), "# Databricks notebook source") == Language.PYTHON
-    assert NotebookLoader.detect_language(Path("hi"), "-- Databricks notebook source") == Language.SQL
-    assert not NotebookLoader.detect_language(Path("hi"), "stuff")
+
+    class NotebookLoaderForTesting(NotebookLoader):
+
+        @classmethod
+        def detect_language(cls, path: Path, content: str):
+            return cls._detect_language(path, content)
+
+    assert NotebookLoaderForTesting.detect_language(Path("hi.py"), "stuff") == Language.PYTHON
+    assert NotebookLoaderForTesting.detect_language(Path("hi.sql"), "stuff") == Language.SQL
+    assert NotebookLoaderForTesting.detect_language(Path("hi"), "# Databricks notebook source") == Language.PYTHON
+    assert NotebookLoaderForTesting.detect_language(Path("hi"), "-- Databricks notebook source") == Language.SQL
+    assert not NotebookLoaderForTesting.detect_language(Path("hi"), "stuff")