Fix update_lib tests (RustPython#6866)

youknowone · web-flow · commit f35791ec64a2 · 2026-01-24T22:47:03.000+09:00
diff --git a/scripts/update_lib/tests/test_deps.py b/scripts/update_lib/tests/test_deps.py
@@ -64,12 +64,19 @@ def test_syntax_error_returns_empty(self):
 class TestGetLibPaths(unittest.TestCase):
     """Tests for get_lib_paths function."""
 
-    def test_known_dependency(self):
-        """Test library with known dependencies."""
-        paths = get_lib_paths("datetime", "cpython")
-        self.assertEqual(len(paths), 2)
-        self.assertIn(pathlib.Path("cpython/Lib/datetime.py"), paths)
-        self.assertIn(pathlib.Path("cpython/Lib/_pydatetime.py"), paths)
+    def test_auto_detect_py_module(self):
+        """Test auto-detection of _py{module}.py pattern."""
+        with tempfile.TemporaryDirectory() as tmpdir:
+            tmpdir = pathlib.Path(tmpdir)
+            lib_dir = tmpdir / "Lib"
+            lib_dir.mkdir()
+            (lib_dir / "datetime.py").write_text("# datetime")
+            (lib_dir / "_pydatetime.py").write_text("# _pydatetime")
+
+            paths = get_lib_paths("datetime", str(tmpdir))
+            self.assertEqual(len(paths), 2)
+            self.assertIn(tmpdir / "Lib" / "datetime.py", paths)
+            self.assertIn(tmpdir / "Lib" / "_pydatetime.py", paths)
 
     def test_default_file(self):
         """Test default to .py file."""
