class StateSaveBufferTestCase(TestCase):
@setup
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)
def test_save(self):
assert self.buffer.save(1, 2)
assert not self.buffer.save(1, 3)
assert not self.buffer.save(1, 4)
assert not self.buffer.save(1, 5)
assert not self.buffer.save(1, 6)
assert self.buffer.save(1, 7)
assert_equal(self.buffer.buffer[1], 7)
def test__iter__(self):
self.buffer.save(1, 2)
self.buffer.save(2, 3)
items = list(self.buffer)
assert not self.buffer.buffer
assert_equal(items, [(1, 2), (2, 3)])
class TestStateSaveBuffer(TestCase):
@setup
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)
def test_save(self):
assert self.buffer.save(1, 2)
assert not self.buffer.save(1, 3)
assert not self.buffer.save(1, 4)
assert not self.buffer.save(1, 5)
assert not self.buffer.save(1, 6)
assert self.buffer.save(1, 7)
assert_equal(self.buffer.buffer[1], 7)
def test__iter__(self):
self.buffer.save(1, 2)
self.buffer.save(2, 3)
items = list(self.buffer)
assert not self.buffer.buffer
assert_equal(items, [(1, 2), (2, 3)])
