Handle save paths across Windows drives
This commit is contained in:
@@ -4,6 +4,7 @@ import sys
|
||||
import tempfile
|
||||
import types
|
||||
import unittest
|
||||
from unittest import mock
|
||||
|
||||
|
||||
class FakeImageArray:
|
||||
@@ -185,6 +186,32 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
self.assertEqual(result["ui"]["images"][0]["type"], "external")
|
||||
self.assertEqual(result["ui"]["images"][0]["subfolder"], external_dir.replace("\\", "/"))
|
||||
|
||||
def test_save_image_handles_windows_different_drive_paths(self):
|
||||
node = self.image_nodes.DumasSaveImageNode()
|
||||
image = FakeTensorBatch()
|
||||
|
||||
with mock.patch.object(
|
||||
self.image_nodes.folder_paths,
|
||||
"get_output_directory",
|
||||
return_value="C:\\ComfyUI\\output",
|
||||
), mock.patch.object(
|
||||
self.image_nodes.os.path,
|
||||
"abspath",
|
||||
return_value="D:\\renders",
|
||||
):
|
||||
result = node.save_images(
|
||||
images=image,
|
||||
folder="D:\\renders",
|
||||
pattern="drive_%counter%",
|
||||
format="png",
|
||||
quality=100,
|
||||
embed_workflow=False,
|
||||
save_on_run=True,
|
||||
)
|
||||
|
||||
self.assertEqual(result["ui"]["images"][0]["type"], "external")
|
||||
self.assertEqual(result["ui"]["images"][0]["subfolder"], "D:/renders")
|
||||
|
||||
def test_save_image_skips_when_save_is_disabled(self):
|
||||
node = self.image_nodes.DumasSaveImageNode()
|
||||
image = FakeTensorBatch()
|
||||
|
||||
Reference in New Issue
Block a user