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.
78 lines
2.1 KiB
78 lines
2.1 KiB
import json |
|
import os |
|
|
|
import pytest |
|
|
|
from clusterview.points import PointSet |
|
from clusterview.point_manager import PointManager |
|
|
|
|
|
@pytest.fixture(autouse=True) |
|
def setup(): |
|
p = PointSet(8, 100, 100) |
|
p.add_point(4, 4) |
|
p.add_point(9, 10) |
|
p.add_point(30, 40) |
|
|
|
PointManager.point_set = p |
|
|
|
def test_load(tmpdir): |
|
test = ("{\n" |
|
"\"point_size\": 8,\n" |
|
"\"viewport_height\": 100,\n" |
|
"\"viewport_width\": 100,\n" |
|
"\"points\":[\n" |
|
"{\"x\": 8, \"y\": 8}," |
|
"{\"x\": 30, \"y\": 50}" |
|
"]\n" |
|
"}") |
|
|
|
p = tmpdir.mkdir("test_data").join("test.json") |
|
p.write(test) |
|
|
|
expected = PointSet(8, 100, 100) |
|
expected.add_point(8, 8) |
|
expected.add_point(30, 50) |
|
|
|
PointManager.load(p) |
|
|
|
print(PointManager.point_set) |
|
|
|
# The fixture point_set inside the singleton PointManager should be |
|
# overwritten. |
|
assert PointManager.point_set == expected |
|
|
|
def test_save(tmpdir): |
|
d = tmpdir.mkdir("test_data").join("save.json") |
|
|
|
PointManager.save(d) |
|
|
|
expected_str = ("{" |
|
"\"point_size\": 8," |
|
"\"viewport_height\": 100," |
|
"\"viewport_width\": 100," |
|
"\"points\":[" |
|
"{\"x\": 4, \"y\": 4}," |
|
"{\"x\": 9, \"y\": 10}," |
|
"{\"x\": 30, \"y\": 40}" |
|
"]" |
|
"}") |
|
|
|
|
|
with open(d) as f: |
|
expected = json.loads(expected_str) |
|
actual = json.load(d) |
|
|
|
# Since the JSON module converts the `points` key to |
|
# a list of dicts, we need to do manual comparison |
|
# to get around the problem of list equality when we |
|
# really want a set of Points. |
|
|
|
assert actual["point_size"] == expected["point_size"] |
|
assert actual["viewport_width"] == expected["viewport_width"] |
|
assert actual["viewport_height"] == expected["viewport_height"] |
|
|
|
assert len(actual["points"]) == len(expected["points"]) |
|
|
|
for p in actual["points"]: |
|
assert p in expected["points"]
|
|
|