def test_has_method(method): """Test that graph has all the correct methods.""" from graph import Graph assert hasattr(ClassDef(), method)