[test] Add InputGraph tests