you should use always the spoiler flag when discussing code (i added the flag to your comment)

same as mine ( :

Yeah, same.

Thanks for your response, there was an edge case where i wasn't evaluating composite number. My apologies for confusion.All good with Kata.

The tests cases in Python are fine:

Python Completions 3709

Please read this: https://docs.codewars.com/training/troubleshooting/

Not a kata issue.

Below is the output from one of the test case , which is broken for python.

[1, 4, 9, 2, 25, 2] should equal [1, 4, 9, 2, 25, 36]

Please correct this.

Thanks akar-0 :)

Tests are ok. And if they were not ok, you don't give the input so nobody could know.

Are tests ok?

expected [ 2, 9, 3, 49, 4, 1, 10, 10201, 25, 25, 1, 1 ] to deeply equal [ 10, 10201, 25, 25, 1, 1 ]

It is working when I test it manually in RunJS i.e.