class TestRestrictionSite(object): '''Test RestrictionSite class.''' def __init__(self): self.ecorv = RestrictionSite(DNA('GATATC'), (3, 3), name='EcoRV') self.foki = RestrictionSite(DNA('GGATG'), (14, 18), name='FokI') def test_cuts_outside(self): '''Test cuts_outside method.''' assert_false(self.ecorv.cuts_outside()) assert_true(self.foki.cuts_outside()) def test_len(self): '''Test len function.''' assert_equal(len(self.ecorv), 6) assert_equal(len(self.foki), 5)