def test_could_tp_mount(mount_type, expected_could_tp):
mount = Mount(type=EntityDBEntry(id=mount_type))
poly_mount = PolyPointer(101, mount, MemContext())
player = Player(overlay=poly_mount)
run_state = RunState()
assert run_state.could_tp(player, set(), set()) == expected_could_tp
def test_could_tp_items(item_set, prev_item_set, expected_could_tp):
run_state = RunState()
assert run_state.could_tp(Player(), item_set,
prev_item_set) == expected_could_tp