Update test_pyamg_install.py