ffxivbis/test/test_piece.py
2019-09-06 00:54:27 +03:00

13 lines
461 B
Python

from service.models.piece import Piece
def test_parse_head(head_with_upgrade: Piece) -> None:
assert Piece.get({'piece': 'head', 'is_tome': True}) == head_with_upgrade
def test_parse_weapon(weapon: Piece) -> None:
assert Piece.get({'piece': 'weapon', 'is_tome': False}) == weapon
def test_parse_upgrade(head_with_upgrade: Piece) -> None:
assert Piece.get({'piece': head_with_upgrade.upgrade.name, 'is_tome': True}) == head_with_upgrade.upgrade