Add unit test module for numpy to compare arrays