od razu zwracamy macierz (trzeba zmienić test.py)
rozmiary danych