81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
import unittest
|
|
import numpy as np
|
|
from pathlib import Path
|
|
import tempfile
|
|
import os
|
|
import sys
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parent.parent
|
|
sys.path.append(str(PROJECT_ROOT.parent))
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
class DummyBase: pass
|
|
mock_types = MagicMock()
|
|
mock_types.NodeSocket = type("NodeSocket", (DummyBase,), {})
|
|
mock_types.Node = type("Node", (DummyBase,), {})
|
|
mock_types.NodeCustomGroup = type("NodeCustomGroup", (DummyBase,), {})
|
|
mock_types.NodeTree = type("NodeTree", (DummyBase,), {})
|
|
mock_types.PropertyGroup = type("PropertyGroup", (DummyBase,), {})
|
|
mock_types.Operator = type("Operator", (DummyBase,), {})
|
|
mock_types.Panel = type("Panel", (DummyBase,), {})
|
|
mock_types.Object = MagicMock
|
|
mock_types.Mesh = MagicMock
|
|
|
|
mock_bpy = MagicMock()
|
|
mock_bpy.types = mock_types
|
|
mock_bpy.path.abspath = lambda path: path.replace("//", "/abs/")
|
|
|
|
sys.modules['bpy'] = mock_bpy
|
|
sys.modules['bpy.types'] = mock_types
|
|
sys.modules['bpy.props'] = MagicMock()
|
|
sys.modules['bpy.utils'] = MagicMock()
|
|
sys.modules['bpy.utils.previews'] = MagicMock()
|
|
sys.modules['bpy.app'] = MagicMock()
|
|
sys.modules['bpy.app.handlers'] = MagicMock()
|
|
|
|
from NodeSlicer.core.nodeslicer.nodes.inputs import SLICENODE_WipeTower, SLICENODE_OffsetObject
|
|
from NodeSlicer.infra.prusaslicer_profiles.models import ConfigWriter
|
|
from NodeSlicer.infra._3mf_writer._3mf import write_model_xml
|
|
from NodeSlicer.infra.blender.blender_mesh_capture import SlicingGroup
|
|
|
|
class TestWipeTowerNode(unittest.TestCase):
|
|
def test_wipe_tower_node_output(self):
|
|
node = SLICENODE_WipeTower()
|
|
node.resolve = lambda key: 180.5 if key == "X Position" else 135.2
|
|
|
|
res = node.evaluate_output("Settings")
|
|
self.assertIn("wipe_tower_position", res)
|
|
self.assertEqual(res["wipe_tower_position"], (180.5, 135.2))
|
|
|
|
def test_offset_object_node_output(self):
|
|
node = SLICENODE_OffsetObject()
|
|
node.resolve = lambda key: {"X Offset": 12.5, "Y Offset": -5.0, "Z Offset": 2.0}.get(key, 0.0)
|
|
|
|
res = node.evaluate_output("Settings")
|
|
self.assertIn("offset_object", res)
|
|
self.assertEqual(res["offset_object"], (12.5, -5.0, 2.0))
|
|
|
|
def test_config_writer_excludes_special_keys(self):
|
|
conf = {
|
|
"layer_height": "0.2",
|
|
"wipe_tower_position": "(180.5, 135.2)",
|
|
"offset_object": "(12.5, -5.0, 2.0)",
|
|
"nozzle_diameter": "0.4"
|
|
}
|
|
writer = ConfigWriter(conf)
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
ini_path = Path(temp_dir) / "Slic3r_PE.config"
|
|
writer.write_ini_3mf(ini_path)
|
|
|
|
with open(ini_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
|
|
self.assertIn("; layer_height = 0.2", content)
|
|
self.assertIn("; nozzle_diameter = 0.4", content)
|
|
self.assertNotIn("wipe_tower_position", content)
|
|
self.assertNotIn("offset_object", content)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|