test_custom_graph_opt in test/unit/tasks/test_custom.py falls in stochastic way by assert. Also it is not realy clear what it tests (there is no docstring). So it should be corrected for more obviousness.