package com.thealgorithms.maths;

import static org.junit.jupiter.api.Assertions.*;

import org.junit.jupiter.api.Test;

class MobiusFunctionTest {

    @Test
    void testMobiusForZero() {
        //given
        int number = 0;
        String expectedMessage = "Number must be greater than zero.";

        //when
        Exception exception = assertThrows(
            IllegalArgumentException.class,
            () -> {
                MobiusFunction.mobius(number);
            }
        );
        String actualMessage = exception.getMessage();

        //then
        assertEquals(expectedMessage, actualMessage);
    }

    @Test
    void testMobiusForNegativeNumber() {
        //given
        int number = -1;
        String expectedMessage = "Number must be greater than zero.";

        //when
        Exception exception = assertThrows(
            IllegalArgumentException.class,
            () -> {
                MobiusFunction.mobius(number);
            }
        );
        String actualMessage = exception.getMessage();

        //then
        assertEquals(expectedMessage, actualMessage);
    }

    @Test
    void testMobiusFunction() {
        int[] expectedResultArray = {
            1,
            -1,
            -1,
            0,
            -1,
            1,
            -1,
            0,
            0,
            1,
            -1,
            0,
            -1,
            1,
            1,
            0,
            -1,
            0,
            -1,
            0,
            1,
            1,
            -1,
            0,
            0,
            1,
            0,
            0,
            -1,
            -1,
            -1,
            0,
            1,
            1,
            1,
            0,
            -1,
            1,
            1,
            0,
            -1,
            -1,
            -1,
            0,
            0,
            1,
            -1,
            0,
            0,
            0,
            1,
            0,
            -1,
            0,
            1,
            0,
            1,
            1,
            -1,
            0,
            -1,
            1,
            0,
            0,
            1,
            -1,
            -1,
            0,
            1,
            -1,
            -1,
            0,
            -1,
            1,
            0,
            0,
            1,
            -1,
            -1,
            0,
            0,
            1,
            -1,
            0,
            1,
            1,
            1,
            0,
            -1,
            0,
            1,
            0,
            1,
            1,
            1,
            0,
            -1,
            0,
            0,
            0,
        };

        for (int i = 1; i <= 100; i++) {
            //given
            int expectedValue = expectedResultArray[i - 1];

            //when
            int actualValue = MobiusFunction.mobius(i);

            //then
            assertEquals(expectedValue, actualValue);
        }
    }
}