Add some extra comments in Graph test method in main.py