import pytest
from mugloar import dragon


def test_partition():
    for solution in dragon.partition(20, 4, 0, 10):
        print(solution)
        assert abs(solution[0]) + abs(solution[1]) + abs(solution[2]) + abs(solution[3]) == 20

import pytest
from mugloar import dragon


@pytest.fixture
def dragon_instance():
    return dragon.Dragon()


@pytest.fixture
def knight():
    return [('endurance', 8), ('attack', 5), ('armor', 4), ('agility', 3)]


@pytest.fixture
def dragon_stats():
    return 10, 10, 0, 0


def test_set_relative_stats(dragon_instance, dragon_stats, knight):
    dragon_instance.set_relative_stats(dragon_stats, knight)


def test_partition():
    for solution in dragon.partition(20, 4, 0, 10):
        assert abs(solution[0]) + abs(solution[1]) + abs(solution[2]) + abs(solution[3]) == 20

Implement rudimentary unit tests for dragon class
Implement rudimentary unit tests for dragon class
