104 lines
2.7 KiB
Python
104 lines
2.7 KiB
Python
import pytest
|
|
|
|
from gaphas import item
|
|
from gaphas.connections import Connections
|
|
from gaphas.constraint import EqualsConstraint
|
|
from gaphas.solver import Solver
|
|
|
|
|
|
@pytest.fixture
|
|
def connections():
|
|
return Connections()
|
|
|
|
|
|
def test_connections_with_custom_solver():
|
|
solver = Solver()
|
|
connections = Connections(solver)
|
|
|
|
assert connections.solver is solver
|
|
|
|
|
|
def test_reconnect_item(connections):
|
|
i = item.Line(connections)
|
|
ii = item.Line(connections)
|
|
|
|
cons1 = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
cons2 = EqualsConstraint(i.handles()[0].pos.y, i.handles()[0].pos.y)
|
|
connections.connect_item(i, i.handles()[0], ii, ii.ports()[0], cons1)
|
|
|
|
assert connections.get_connection(i.handles()[0]).constraint is cons1
|
|
assert cons1 in connections.solver.constraints
|
|
|
|
connections.reconnect_item(i, i.handles()[0], constraint=cons2)
|
|
|
|
assert connections.get_connection(i.handles()[0]).constraint is cons2
|
|
assert cons1 not in connections.solver.constraints
|
|
assert cons2 in connections.solver.constraints
|
|
|
|
|
|
def test_add_item_constraint(connections):
|
|
i = item.Line(connections)
|
|
c1 = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
|
|
connections.add_constraint(i, c1)
|
|
|
|
cinfo = next(connections.get_connections(item=i))
|
|
|
|
assert cinfo.item is i
|
|
assert cinfo.constraint is c1
|
|
|
|
|
|
def test_remove_item_constraint(connections):
|
|
i = item.Line(connections)
|
|
c1 = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
|
|
connections.add_constraint(i, c1)
|
|
connections.remove_constraint(i, c1)
|
|
|
|
assert list(connections.get_connections(item=i)) == []
|
|
|
|
|
|
def test_remove_item_constraint_when_item_is_disconnected(connections):
|
|
i = item.Line(connections)
|
|
c1 = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
|
|
connections.add_constraint(i, c1)
|
|
connections.disconnect_item(i)
|
|
|
|
assert list(connections.get_connections(item=i)) == []
|
|
|
|
|
|
def test_notify_on_constraint_solved(connections):
|
|
events = []
|
|
|
|
def on_notify(cinfo):
|
|
events.append(cinfo)
|
|
|
|
i = item.Line(connections)
|
|
c = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
connections.add_constraint(i, c)
|
|
|
|
connections.add_handler(on_notify)
|
|
connections.solve()
|
|
|
|
assert events
|
|
assert events[0].constraint is c
|
|
|
|
|
|
def test_connection_remove_handler(connections):
|
|
events = []
|
|
|
|
def on_notify(cinfo):
|
|
events.append(cinfo)
|
|
|
|
i = item.Line(connections)
|
|
c = EqualsConstraint(i.handles()[0].pos.x, i.handles()[0].pos.x)
|
|
connections.add_constraint(i, c)
|
|
|
|
connections.add_handler(on_notify)
|
|
connections.remove_handler(on_notify)
|
|
connections.remove_handler(on_notify)
|
|
connections.solve()
|
|
|
|
assert not events
|