import random
import unittest
from beamboot.hf_decoder_memory import DecoderMemory


class DecoderMemoryTests(unittest.TestCase):
    def test_reads_and_overwrites(self):
        memory = DecoderMemory(max_writes=64)
        rng = random.Random(771)
        reference = [0] * 16
        for _ in range(64):
            address, value = rng.randrange(16), rng.randrange(16)
            memory.write(address, value)
            reference[address] = value
            self.assertEqual(memory.read(address), value)
        for address in range(16): self.assertEqual(memory.read(address), reference[address])