#NEWLINE# def test_trace( ) : #NEWLINE# #INDENT# a = [ [ ZZ ( 3 ) , ZZ ( 7 ) , ZZ ( 4 ) ] , [ ZZ ( 2 ) , ZZ ( 4 ) , ZZ ( 5 ) ] , [ ZZ ( 6 ) , ZZ ( 2 ) , ZZ ( 3 ) ] ] #NEWLINE# b = eye ( 2 , ZZ ) #NEWLINE# assert ( trace ( b , ZZ ) == ZZ ( 10 ) ) #NEWLINE# assert ( trace ( b , ZZ ) == ZZ ( 2 ) )