def testBlacklistSet(self): blset = utils.BlacklistSet(0.1) blset.update([1, 2]) time.sleep(0.3) blset.add(3) with self.assertRaises(KeyError): blset.remove(2) self.assertNotIn(1, blset) self.assertNotIn(2, blset) self.assertIn(3, blset) blset.add(4) time.sleep(0.3) blset.add(5) blset.add(6) self.assertSetEqual({5, 6}, set(blset))
def test_blacklist_set(): blset = utils.BlacklistSet(0.1) blset.update([1, 2]) time.sleep(0.3) blset.add(3) with pytest.raises(KeyError): blset.remove(2) assert 1 not in blset assert 2 not in blset assert 3 in blset blset.add(4) time.sleep(0.3) blset.add(5) blset.add(6) assert {5, 6} == set(blset)