Testing out implementations of graph algorithms in python