def test2(): assert "needle" in assignment.function() assert "needle2" in assignment.function(),"error Message"
def test1(): assert assignment.function() == expected output