add the Python test script for matrix multipilcation