Test optimizer with polymorphic relay node

[stygmate]

Add node polymorphism tests and related schema
Update asserts in optimizer.py -> To be double checked before merge

[stygmate]

@bellini666 here it is :)

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 88.94%. Comparing base (7b8a6a2) to head (fb40e9f).
Report is 7 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #570      +/-   ##
==========================================
+ Coverage   88.81%   88.94%   +0.13%     
==========================================
  Files          41       41              
  Lines        3531     3547      +16     
==========================================
+ Hits         3136     3155      +19     
+ Misses        395      392       -3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[bellini666]

Thanks @stygmate :)

I had to do a small modification from isinstance(foo, A | B) to isinstance(foo, (A, B)) as the new syntax is only compatible with python 3.10+, and we are still supporting 3.8/3.9