modify test.py and rename as three_d_array.py
'