please display the running of your python code. You will need to create two text files

Write a python function identityMatrix(rank) that takes as input a number rank and RETURNS a 2D array that is a square identity matrix of that rank. The rank of a NxN square matrix is N. An identity matrix has 1s along the diagonal and 0s everywhere else. This is a 5x5, or rank 5, identity matrix: 1 0 0 0 0 0 1 0 0 0 0 0 1 0 0 0 0 0 1 0 0 0 0 0 1

>>> identityMatrix(1)