Split unit test into examples vs plotting routines

[joanna-pk]

Currently the main unit test fails, but only due to network graph plotting notebook. Torch vs onnx compatibility is an issue.