Files
pytiled_parser/tests/test_tileset.py
2021-12-19 22:23:09 -05:00

73 lines
2.3 KiB
Python

"""Tests for tilesets"""
import importlib.util
import json
import os
import xml.etree.ElementTree as etree
from pathlib import Path
import pytest
from pytiled_parser.common_types import OrderedPair, Size
from pytiled_parser.parsers.json.tileset import parse as parse_json
from pytiled_parser.parsers.tmx.tileset import parse as parse_tmx
TESTS_DIR = Path(os.path.dirname(os.path.abspath(__file__)))
TEST_DATA = TESTS_DIR / "test_data"
TILE_SETS = TEST_DATA / "tilesets"
ALL_TILESET_DIRS = [
TILE_SETS / "image",
TILE_SETS / "image_background_color",
TILE_SETS / "image_grid",
TILE_SETS / "image_properties",
TILE_SETS / "image_transparent_color",
TILE_SETS / "image_tile_offset",
TILE_SETS / "image_transformations",
TILE_SETS / "individual_images",
TILE_SETS / "terrain",
]
def fix_object(my_object):
my_object.coordinates = OrderedPair(
round(my_object.coordinates[0], 4), round(my_object.coordinates[1], 4)
)
my_object.size = Size(round(my_object.size[0], 4), round(my_object.size[1], 4))
def fix_tileset(tileset):
tileset.version = None
tileset.tiled_version = None
if tileset.tiles:
for tile in tileset.tiles.values():
if tile.objects:
for my_object in tile.objects.tiled_objects:
fix_object(my_object)
@pytest.mark.parametrize("parser_type", ["json", "tmx"])
@pytest.mark.parametrize("tileset_dir", ALL_TILESET_DIRS)
def test_tilesets_integration(parser_type, tileset_dir):
# it's a PITA to import like this, don't do it
# https://stackoverflow.com/a/67692/1342874
spec = importlib.util.spec_from_file_location(
"expected", tileset_dir / "expected.py"
)
expected = importlib.util.module_from_spec(spec)
spec.loader.exec_module(expected)
if parser_type == "json":
raw_tileset_path = tileset_dir / "tileset.json"
with open(raw_tileset_path) as raw_tileset:
tileset_ = parse_json(json.loads(raw_tileset.read()), 1)
elif parser_type == "tmx":
raw_tileset_path = tileset_dir / "tileset.tsx"
with open(raw_tileset_path) as raw_tileset:
tileset_ = parse_tmx(etree.parse(raw_tileset).getroot(), 1)
fix_tileset(tileset_)
fix_tileset(expected.EXPECTED)
assert tileset_ == expected.EXPECTED