def test_name_index_equality(): db1 = ScreedSequenceDB(testfa) db2 = ScreedSequenceDB_ByIndex(testfa) # must use something other than the obj itself for comparison... v1 = sorted([ (v.name, v.seq) for v in db1.itervalues() ]) v2 = sorted([ (v.name, v.seq) for v in db2.itervalues() ]) assert v1 == v2, (v1, v2)
def test_seqinfodict_by_index(): db1 = ScreedSequenceDB_ByIndex(testfa) sd = db1.seqInfoDict m = sorted([ x for (x, y) in sd.iteritems() ]) n = sorted([ x for x in sd.iterkeys() ]) assert m == n, (m, n)
def test_pickle_ByIndex(): db = ScreedSequenceDB_ByIndex(testfa) ofp = StringIO() dump(db, ofp) ifp = StringIO(ofp.getvalue()) db2 = load(ifp) assert db.filepath == db2.filepath
def test_index_iterator_methods(): db = ScreedSequenceDB_ByIndex(testfa) # test the various iterator methods for equal results from db m = sorted([ (x, db[x]) for x in db ]) n = sorted([ i for i in db.iteritems() ]) o = sorted([ (v.record.id, v) for v in db.itervalues() ]) assert m == n assert m == o, (m, o)