Use assert_allclose instead of assert_equal when testing functions. This test is not about mathematical precision but that the correct functions are computed.