You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.2 KiB
52 lines
1.2 KiB
import pytest |
|
|
|
from clusterview.points import Attribute, PointSet |
|
|
|
|
|
def test_add_to_point_set(): |
|
l = PointSet() |
|
|
|
l.add_point(1, 2) |
|
|
|
points = list(l.points) |
|
|
|
assert len(points) == 1 |
|
assert points[0] == (1, 2) |
|
assert len(l.attributes(1, 2)) == 0 |
|
|
|
def test_add_to_point_set_with_attributes(): |
|
attribute = Attribute("thing", 1) |
|
|
|
l = PointSet() |
|
l.add_point(1, 2, attrs=[attribute]) |
|
|
|
points = list(l.points) |
|
attrs = l.attributes(1, 2) |
|
|
|
assert len(points) == 1 |
|
assert points[0] == (1, 2) |
|
assert len(l.attributes(1, 2)) == 1 |
|
|
|
def test_remove_point(): |
|
attribute = Attribute("thing", 1) |
|
|
|
l = PointSet() |
|
l.add_point(1, 2, attrs=[attribute]) |
|
|
|
l.remove_point(1, 2) |
|
|
|
points = list(l.points) |
|
|
|
assert len(points) == 0 |
|
|
|
with pytest.raises(KeyError): |
|
# We expect a call to attributes on a removed |
|
# point to raise a KeyError because it no |
|
# longer exists in the point -> attribute_list |
|
# dictionary. |
|
l.attributes(1, 2) |
|
|
|
def test_attributes_must_be_array_of_attributes(): |
|
with pytest.raises(ValueError): |
|
l = PointSet() |
|
l.add_point(1, 2, attrs=[1,2,3,4,5])
|
|
|