Transfer Matrix Method Python Code Tester Step