Fix test answer error
Merge pull request #3 from ChrisBeaumont/fix-assert-allclose

Fix test answer error