fix correct superclass for IDir (#356)

insitro · Apr 16, 2024 · 132bcf1 · 132bcf1
1 parent 37df29f
commit 132bcf1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/redun/file.py b/redun/file.py
@@ -1677,7 +1677,7 @@ class IStagingFile(StagingFile):
     classes = IFileClasses()
 
 
-class IStagingDir(Staging[Dir]):
+class IStagingDir(StagingDir):
     type_basename = "IStagingDir"
     type_name = "redun.IStagingDir"
     classes = IFileClasses()
@@ -1738,7 +1738,7 @@ class ContentStagingFile(StagingFile):
     classes = ContentFileClasses()
 
 
-class ContentStagingDir(Staging[Dir]):
+class ContentStagingDir(StagingDir):
     type_basename = "ContentStagingDir"
     type_name = "redun.ContentStagingDir"
     classes = ContentFileClasses()

diff --git a/redun/tests/test_file.py b/redun/tests/test_file.py
@@ -1017,6 +1017,11 @@ def test_ifile_classes() -> None:
     IFile("dir/b").touch()
     assert all(isinstance(file, IFile) for file in IDir("dir"))
 
+    assert IFile("dir/a").stage("a").render_stage() == "cp dir/a a"
+    assert IFile("dir/a").stage("a").render_unstage() == "cp a dir/a"
+    assert IDir("dir").stage("dir2").render_stage() == "cp -r dir dir2"
+    assert IDir("dir").stage("dir2").render_unstage() == "cp -r dir2 dir"
+
 
 @use_tempdir
 def test_content_file() -> None: