mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 05:39:37 +08:00
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
from UM.Math.Polygon import Polygon
|
|
from UM.Scene.SceneNodeDecorator import SceneNodeDecorator
|
|
from cura.Scene.CuraSceneNode import CuraSceneNode
|
|
import pytest
|
|
|
|
from unittest.mock import patch
|
|
|
|
|
|
class MockedConvexHullDecorator(SceneNodeDecorator):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def getConvexHull(self):
|
|
return Polygon([[5, 5], [-5, 5], [-5, -5], [5, -5]])
|
|
|
|
def getPrintingArea(self):
|
|
return Polygon([[5, 5], [-5, 5], [-5, -5], [5, -5]])
|
|
|
|
|
|
class InvalidConvexHullDecorator(SceneNodeDecorator):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def getConvexHull(self):
|
|
return Polygon()
|
|
|
|
|
|
@pytest.fixture()
|
|
def cura_scene_node():
|
|
# Replace the SettingOverrideDecorator with an empty decorator
|
|
with patch("cura.Scene.CuraSceneNode.SettingOverrideDecorator", SceneNodeDecorator):
|
|
return CuraSceneNode()
|
|
|
|
|
|
class TestCollidesWithAreas:
|
|
def test_noConvexHull(self, cura_scene_node):
|
|
assert not cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])
|
|
|
|
def test_convexHullIntersects(self, cura_scene_node):
|
|
cura_scene_node.addDecorator(MockedConvexHullDecorator())
|
|
assert cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])
|
|
|
|
def test_convexHullNoIntersection(self, cura_scene_node):
|
|
cura_scene_node.addDecorator(MockedConvexHullDecorator())
|
|
|
|
assert not cura_scene_node.collidesWithAreas([Polygon([[60, 60], [40, 60], [40, 40], [60, 40]])])
|
|
|
|
def test_invalidConvexHull(self, cura_scene_node):
|
|
cura_scene_node.addDecorator(InvalidConvexHullDecorator())
|
|
assert not cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])
|
|
|
|
|
|
def test_outsideBuildArea(cura_scene_node):
|
|
cura_scene_node.setOutsideBuildArea(True)
|
|
assert cura_scene_node.isOutsideBuildArea
|