Clean tests

[fandreuz]

It would be nice to organize better the test cases. Maybe using numpy (or something similar) to store graphs(?)

[fandreuz]

Someone on Reddit suggested https://hypothesis.readthedocs.io/en/latest/index.html