added coverage and first unittest for graph wrapper