Modified tests to allow naive python implementation