Move test_pyamgcl out of module