Fix lap_graph.py comments and update unit tests