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_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)