Fix solve tests in test_wester.py