from gpt_engineer.db import DB


def test_db():
    # use /tmp for testing
    db = DB('/tmp/test_db')
    db['test'] = 'test'
    assert db['test'] == 'test'
    db['test'] = 'test2'
    assert db['test'] == 'test2'
    db['test2'] = 'test2'
    assert db['test2'] == 'test2'
    assert db['test'] == 'test2'
    print('test_db passed')