Use astcheck for tests

See https://github.com/takluyver/astcheck