import pypuppetdb


def test_connect_api():
    puppetdb = pypuppetdb.connect()
    assert puppetdb.version == 'v4'

@@ -109,8 +109,128 @@
 == 'v4'%0A
+%0A%0Adef test_connect_with_statement():%0A    with pypuppetdb.connect() as puppetdb:%0A        assert puppetdb.version == 'v4'%0A

Add test for creating connection with 'with' statement.