import pytest
import numpy as np

from sudoku.symetries import mat_sym


def test_same_output_under_rotation():
    assert mat_sym.shape == (9 * 9 * 9, 16, 9 * 9 * 9)

    arr1 = np.zeros((9, 9, 9))
    arr1[0, 1, 2] = 1
    arr1_sym = np.dot(mat_sym, arr1.reshape(9 * 9 * 9, -1)).reshape(9, 9, 9, -1)
    arr2 = np.zeros((9, 9, 9))
    arr2[1, 2, 3] = 1
    arr2_sym = np.dot(mat_sym, arr2.reshape(9 * 9 * 9, -1)).reshape(9, 9, 9, -1)
    assert arr1_sym.shape == (9, 9, 9, 16)
    assert (arr1_sym[0, 1, 2] == arr2_sym[1, 2, 3]).all()
    assert (arr1_sym[5, 1, 2] == arr2_sym[5, 2, 3]).all()
    assert (arr1_sym[0, 1, 1] == arr2_sym[1, 2, 4]).all()
    assert (arr1_sym[0, 5, 1] == arr2_sym[1, 8, 4]).all()
    assert (arr1_sym[6, 5, 1] == arr2_sym[6, 8, 4]).all()