initial commit

This commit is contained in:
2019-09-06 00:54:27 +03:00
commit 9f19519b75
47 changed files with 2027 additions and 0 deletions

20
test/test_bis.py Normal file
View File

@ -0,0 +1,20 @@
from service.models.bis import BiS
from service.models.piece import Piece
from service.models.upgrade import Upgrade
def test_set_item(bis: BiS, weapon: Piece) -> None:
bis.set_item(weapon)
assert bis.has_piece(weapon)
def test_remove_item(bis: BiS, weapon: Piece) -> None:
test_set_item(bis, weapon)
bis.remove_item(weapon)
assert not bis.has_piece(weapon)
def test_upgrades_required(bis: BiS, weapon: Piece, head_with_upgrade: Piece) -> None:
bis.set_item(weapon)
bis.set_item(head_with_upgrade)
assert bis.upgrades_required == {Upgrade.NoUpgrade: 1, Upgrade.GearUpgrade: 1}