Transfer Matrix Method Python Code Tester