Fix tests to use numpy's assert_almost_equal