Python Matrix class -  Add testcases