from yatzy import Yatzy


# These unit tests can be run using the py.test framework
# available from http://pytest.org/

def test_chance_scores_sum_of_all_dice():
    expected = 15
    actual = Yatzy.chance(2, 3, 4, 5, 1)
    assert expected == actual
    assert 16 == Yatzy.chance(3, 3, 4, 5, 1)


def test_yatzy_scores_50():
    expected = 50
    actual = Yatzy.yatzy([4, 4, 4, 4, 4])
    assert expected == actual
    assert 50 == Yatzy.yatzy([6, 6, 6, 6, 6])
    assert 0 == Yatzy.yatzy([6, 6, 6, 6, 3])


def test_1s():
    assert Yatzy.ones(1, 2, 3, 4, 5) == 1
    assert 2 == Yatzy.ones(1, 2, 1, 4, 5)
    assert 0 == Yatzy.ones(6, 2, 2, 4, 5)
    assert 4 == Yatzy.ones(1, 2, 1, 1, 1)


def test_2s():
    assert 4 == Yatzy.twos(1, 2, 3, 2, 6)
    assert 10 == Yatzy.twos(2, 2, 2, 2, 2)


def test_threes():
    assert 6 == Yatzy.threes(1, 2, 3, 2, 3)
    assert 12 == Yatzy.threes(2, 3, 3, 3, 3)


def test_fours_test():
    assert 12 == Yatzy(4, 4, 4, 5, 5).fours()
    assert 8 == Yatzy(4, 4, 5, 5, 5).fours()
    assert 4 == Yatzy(4, 5, 5, 5, 5).fours()


def test_fives():
    assert 10 == Yatzy(4, 4, 4, 5, 5).fives()
    assert 15 == Yatzy(4, 4, 5, 5, 5).fives()
    assert 20 == Yatzy(4, 5, 5, 5, 5).fives()


def test_sixes_test():
    assert 0 == Yatzy(4, 4, 4, 5, 5).sixes()
    assert 6 == Yatzy(4, 4, 6, 5, 5).sixes()
    assert 18 == Yatzy(6, 5, 6, 6, 5).sixes()


def test_one_pair():
    assert 6 == Yatzy.score_pair(3, 4, 3, 5, 6)
    assert 10 == Yatzy.score_pair(5, 3, 3, 3, 5)
    assert 12 == Yatzy.score_pair(5, 3, 6, 6, 5)


def test_two_Pair():
    assert 16 == Yatzy.two_pair(3, 3, 5, 4, 5)
    assert 18 == Yatzy.two_pair(3, 3, 6, 6, 6)
    assert 0 == Yatzy.two_pair(3, 3, 6, 5, 4)


def test_three_of_a_kind():
    assert 9 == Yatzy.three_of_a_kind(3, 3, 3, 4, 5)
    assert 15 == Yatzy.three_of_a_kind(5, 3, 5, 4, 5)
    assert 9 == Yatzy.three_of_a_kind(3, 3, 3, 3, 5)


def test_four_of_a_knd():
    assert 12 == Yatzy.four_of_a_kind(3, 3, 3, 3, 5)
    assert 20 == Yatzy.four_of_a_kind(5, 5, 5, 4, 5)
    assert 12 == Yatzy.four_of_a_kind(3, 3, 3, 3, 3)
    assert 0 == Yatzy.four_of_a_kind(3, 3, 3, 2, 1)


def test_smallStraight():
    assert 15 == Yatzy.smallStraight(1, 2, 3, 4, 5)
    assert 15 == Yatzy.smallStraight(2, 3, 4, 5, 1)
    assert 0 == Yatzy.smallStraight(1, 2, 2, 4, 5)


def test_largeStraight():
    assert 20 == Yatzy.largeStraight(6, 2, 3, 4, 5)
    assert 20 == Yatzy.largeStraight(2, 3, 4, 5, 6)
    assert 0 == Yatzy.largeStraight(1, 2, 2, 4, 5)


def test_fullHouse():
    assert 18 == Yatzy.fullHouse(6, 2, 2, 2, 6)
    assert 0 == Yatzy.fullHouse(2, 3, 4, 5, 6)

if __name__ == '__main__':
    try:
        test_chance_scores_sum_of_all_dice()
    except Exception as e:
        print(e)
    else:
        print("Test 1 worked!")

    try:
        test_yatzy_scores_50()
    except Exception as e:
        print(e)
    else:
        print("Test 2 worked!")

    try:
        test_1s()
    except Exception as e:
        print(e)
    else:
        print("Test 3 worked!")

    try:
        test_2s()
    except Exception as e:
        print(e)
    else:
        print("Test 4 worked!")
        
    try:
        test_threes()
    except Exception as e:
        print(e)
    else:
        print("Test 5 worked!")

    try:
        test_fours_test()
    except Exception as e:
        print(e)
    else:
        print("Test 6 worked!")

    try:
        test_fives()
    except Exception as e:
        print(e)
    else:
        print("Test 7 worked!")

    try:
        test_sixes_test()
    except Exception as e:
        print(e)
    else:
        print("Test 8 worked!")

    try:
        test_one_pair()
    except Exception as e:
        print(e)
    else:
        print("Test 9 worked!")

    try:
        test_two_Pair()
    except Exception as e:
        print(e)
    else:
        print("Test 10 worked!")

    try:
        test_three_of_a_kind()
    except Exception as e:
        print(e)
    else:
        print("Test 11 worked!")

    try:
        test_four_of_a_knd()
    except Exception as e:
        print(e)
    else:
        print("Test 12 worked!")

    try:
        test_smallStraight()
    except Exception as e:
        print(e)
    else:
        print("Test 13 worked!")

    try:
        test_largeStraight()
    except Exception as e:
        print(e)
    else:
        print("Test 14 worked!")

    try:
        test_fullHouse()
    except Exception as e:
        print(e)
    else:
        print("Test 15 worked!")