#NEWLINE# def test_sum_of_three_squares( ) : #NEWLINE# #INDENT# for i in [ 0 , 1 , 2 , 34 , 123 , 34304595905 , 34304595905394941 , 343045959052344 , 800 , 801 , 802 , 803 , 804 , 805 , 806 ] : #NEWLINE# #INDENT# ( a , b , c ) = sum_of_three_squares ( i ) #NEWLINE# assert ( ( ( ( a ** 2 ) + ( b ** 2 ) ) + ( c ** 2 ) ) == i ) #NEWLINE# #UNINDENT# assert ( sum_of_three_squares ( 7 ) is None ) #NEWLINE# assert ( sum_of_three_squares ( ( ( 4 ** 5 ) * 15 ) ) is None ) #NEWLINE# assert ( sum_of_three_squares ( 25 ) == ( 5 , 0 , 0 ) ) #NEWLINE# assert ( sum_of_three_squares ( 4 ) == ( 0 , 0 , 2 ) )