def test_bad_add_set_function(self, initdb, tprimers):
'''Should raise errors if invalid primers supplied.'''
with pytest.raises(swga.SWGAError):
database.add_set(_id=2, primers=None, score=100)
with pytest.raises(swga.SWGAError):
invalid_primers = Primer.select().where(Primer.seq == "XX")
database.add_set(_id=3, primers=invalid_primers, score=100)
def test_add_set_function(self, initdb, tprimers):
'''Tests adding a set using add_set function.'''
s = database.add_set(_id=2, primers=tprimers, score=100)
assert s.primers.count() == len(tprimers)
for primer in s.primers:
assert primer in tprimers
