代码之家  ›  专栏  ›  技术社区  ›  Piotr Byzia

动态创建对象的Doctest

  •  3
  • Piotr Byzia  · 技术社区  · 17 年前

    def get_session(db_name, verbose, test):
    """Returns current DB session from SQLAlchemy pool.
    
    >>> get_session('Mmusc20090126', False, True)
    <sqlalchemy.orm.session.Session object at 0xfb5ff0>
    
    """
    if test:
        engine = create_engine('sqlite:///:memory:', echo=verbose)
        log_load.debug('DB in RAM.')
    else:
        engine = create_engine('sqlite:///' + 'DB/' + db_name + '.db', echo=verbose)
        log_load.debug('DB stored in file: %s' % 'DB/' + db_name + '.db')
    
    # Create TABLES: Structures, Interactions, Interactors, PDB_UniProt, UniProtSeq
    meta.create_all(engine)
    
    Session = sessionmaker(bind=engine)
    session = Session()
    
    return session
    
    1 回复  |  直到 17 年前
        1
  •  9
  •   DNS    17 年前

    我想你应该用省略号,像这样:

    >>> get_session('Mmusc20090126', False, True) #doctest: +ELLIPSIS
    <sqlalchemy.orm.session.Session object at 0x...>
    

    看见 here