make localFS mv() respect auto_mkdir (#1957)

Author: cccs-malik

fsspec/implementations/local.py

@@ -166,6 +166,10 @@ def mv(self, path1, path2, recursive: bool = True, **kwargs):
         """
         path1 = self._strip_protocol(path1)
         path2 = self._strip_protocol(path2)
+
+        if self.auto_mkdir:
+            self.makedirs(self._parent(path2), exist_ok=True)
+
         shutil.move(path1, path2)
 
     def link(self, src, dst, **kwargs):

fsspec/implementations/tests/test_local.py

@@ -1043,6 +1043,24 @@ def test_mv_recursive(tmpdir):
     assert localfs.info(os.path.join(dest, "afile"))
 
 
+def test_mv_auto_mkdir(tmpdir):
+    localfs = fsspec.filesystem("file", auto_mkdir=True)
+    src = os.path.join(str(tmpdir), "src")
+    dest_dir = os.path.join(str(tmpdir), "dest_dir")
+    dest = os.path.join(dest_dir, "dest")
+
+    assert localfs.exists(src) is False
+    assert localfs.exists(dest) is False
+    assert localfs.exists(dest_dir) is False
+
+    localfs.touch(src)
+    localfs.mv(src, dest)
+
+    assert localfs.exists(src) is False
+    assert localfs.isfile(dest)
+    assert localfs.info(dest)
+
+
 @pytest.mark.xfail(WIN, reason="windows expand path to be revisited")
 def test_copy_errors(tmpdir):
     localfs = fsspec.filesystem("file", auto_mkdir=True)