Fix test answer error

It turns out the "reference" arrays were computed incorrectly, and np.testing.assert_array_almost_equal didn't complain about a difference between two tiny arrays.