Python setUp示例

示例#1

文件： test_mquery_db.py 项目： agroszer/pjpersist

def setUp(test):
    testing.setUp(test)

    conn = test.globs['conn']
    with conn.cursor() as cur:
        cur.execute("CREATE TABLE mq (id SERIAL PRIMARY KEY, data JSONB)")
        cur.execute("INSERT INTO mq (data) VALUES " +
                    ", ".join(jformat(datum) for datum in dataset))
    conn.commit()

示例#2

文件： test_mquery_db.py 项目： sjustas/pjpersist

def setUp(test):
    testing.setUp(test)

    conn = test.globs['conn']
    with conn.cursor() as cur:
        cur.execute("CREATE TABLE mq (id SERIAL PRIMARY KEY, data JSONB)")
        cur.execute("INSERT INTO mq (data) VALUES " +
                    ", ".join(jformat(datum) for datum in dataset))
    conn.commit()

示例#3

def setUp(test):
    testing.setUp(test)

    conn = test.globs['conn']
    with conn.cursor() as cur:
        cur.execute("CREATE TABLE ser (id SERIAL PRIMARY KEY, data JSONB)")
        for dt in DTIMES:
            obj = dict(dtime=dt)
            cur.execute("INSERT INTO ser (data) VALUES (%s)", (jformat(obj), ))
    conn.commit()

示例#4

文件： test_serialize_db.py 项目： Shoobx/pjpersist

def setUp(test):
    testing.setUp(test)

    conn = test.globs['conn']
    with conn.cursor() as cur:
        cur.execute("CREATE TABLE ser (id SERIAL PRIMARY KEY, data JSONB)")
        for dt in DTIMES:
            obj = dict(dtime=dt)
            cur.execute("INSERT INTO ser (data) VALUES (%s)", (jformat(obj),))
    conn.commit()

示例#5

文件： test_doc.py 项目： Shoobx/pjpersist

def setUp(test):
    testing.setUp(test)
    # silence this, otherwise half-baked objects raise exceptions
    # on trying to __repr__ missing attributes
    test.orig_DEBUG_EXCEPTION_FORMATTER = \
        exceptionformatter.DEBUG_EXCEPTION_FORMATTER
    exceptionformatter.DEBUG_EXCEPTION_FORMATTER = 0

    def fetchone(table, clause=None):
        qry = 'SELECT * FROM ' + table
        if clause:
            qry += ' WHERE ' + clause
        with test.globs['dm'].getCursor(False) as cur:
            cur.execute(qry)
            return cur.fetchone()
    test.globs['fetchone'] = fetchone

示例#6

def setUp(test):
    testing.setUp(test)
    # silence this, otherwise half-baked objects raise exceptions
    # on trying to __repr__ missing attributes
    test.orig_DEBUG_EXCEPTION_FORMATTER = \
        exceptionformatter.DEBUG_EXCEPTION_FORMATTER
    exceptionformatter.DEBUG_EXCEPTION_FORMATTER = 0

    def fetchone(table, clause=None):
        qry = 'SELECT * FROM ' + table
        if clause:
            qry += ' WHERE ' + clause
        with test.globs['dm'].getCursor(False) as cur:
            cur.execute(qry)
            return cur.fetchone()

    test.globs['fetchone'] = fetchone