|
| 1 | +# nmigen: UnusedElaboratable=no |
| 2 | + |
| 3 | +import unittest |
| 4 | +from nmigen import * |
| 5 | +from nmigen.back.pysim import * |
| 6 | + |
| 7 | +from ..csr import * |
| 8 | +from .. import event |
| 9 | + |
| 10 | + |
| 11 | +def simulation_test(dut, process): |
| 12 | + with Simulator(dut, vcd_file=open("test.vcd", "w")) as sim: |
| 13 | + sim.add_clock(1e-6) |
| 14 | + sim.add_sync_process(process) |
| 15 | + sim.run() |
| 16 | + |
| 17 | + |
| 18 | +class EventMonitorTestCase(unittest.TestCase): |
| 19 | + def test_params(self): |
| 20 | + monitor = EventMonitor(data_width=16, alignment=4, trigger="rise") |
| 21 | + self.assertEqual(monitor.bus.data_width, 16) |
| 22 | + self.assertEqual(monitor.bus.memory_map.alignment, 4) |
| 23 | + self.assertEqual(monitor.src.trigger, event.Source.Trigger.RISE) |
| 24 | + |
| 25 | + def test_trigger_wrong(self): |
| 26 | + with self.assertRaisesRegex(ValueError, |
| 27 | + r"Invalid trigger mode 'foo'; must be one of level, rise, fall"): |
| 28 | + EventMonitor(data_width=8, trigger="foo") |
| 29 | + |
| 30 | + def test_add(self): |
| 31 | + monitor = EventMonitor(data_width=8) |
| 32 | + sub = event.Source() |
| 33 | + monitor.add(sub) |
| 34 | + self.assertEqual(monitor.src.event_map.size, 1) |
| 35 | + self.assertEqual(monitor.src.event_map.index(sub), 0) |
| 36 | + |
| 37 | + def test_freeze(self): |
| 38 | + monitor = EventMonitor(data_width=8) |
| 39 | + monitor.freeze() |
| 40 | + sub = event.Source() |
| 41 | + with self.assertRaisesRegex(ValueError, |
| 42 | + r"Event map has been frozen. Cannot add source."): |
| 43 | + monitor.add(sub) |
| 44 | + |
| 45 | + def test_src_freeze(self): |
| 46 | + monitor = EventMonitor(data_width=8) |
| 47 | + monitor.src |
| 48 | + sub = event.Source() |
| 49 | + with self.assertRaisesRegex(ValueError, |
| 50 | + r"Event map has been frozen. Cannot add source."): |
| 51 | + monitor.add(sub) |
| 52 | + |
| 53 | + def test_bus_freeze(self): |
| 54 | + monitor = EventMonitor(data_width=8) |
| 55 | + monitor.bus |
| 56 | + sub = event.Source() |
| 57 | + with self.assertRaisesRegex(ValueError, |
| 58 | + r"Event map has been frozen. Cannot add source."): |
| 59 | + monitor.add(sub) |
| 60 | + |
| 61 | + def test_csr_regs(self): |
| 62 | + monitor = EventMonitor(data_width=8) |
| 63 | + sub_0 = event.Source() |
| 64 | + sub_1 = event.Source() |
| 65 | + monitor.add(sub_0) |
| 66 | + monitor.add(sub_1) |
| 67 | + resources = list(monitor.bus.memory_map.resources()) |
| 68 | + self.assertEqual(len(resources), 2) |
| 69 | + enable, enable_range = resources[0] |
| 70 | + pending, pending_range = resources[1] |
| 71 | + self.assertEqual( |
| 72 | + (enable.width, enable.access, enable_range), |
| 73 | + (2, Element.Access.RW, (0, 1)) |
| 74 | + ) |
| 75 | + self.assertEqual( |
| 76 | + (pending.width, pending.access, pending_range), |
| 77 | + (2, Element.Access.RW, (1, 2)) |
| 78 | + ) |
| 79 | + |
| 80 | + def test_freeze_idempotent(self): |
| 81 | + monitor = EventMonitor(data_width=8) |
| 82 | + src = monitor.src |
| 83 | + bus = monitor.bus |
| 84 | + monitor.freeze() |
| 85 | + self.assertIs(src, monitor.src) |
| 86 | + self.assertIs(bus, monitor.bus) |
| 87 | + |
| 88 | + |
| 89 | +class EventMonitorSimulationTestCase(unittest.TestCase): |
| 90 | + def test_simple(self): |
| 91 | + dut = EventMonitor(data_width=8) |
| 92 | + sub = event.Source() |
| 93 | + dut.add(sub) |
| 94 | + |
| 95 | + addr_enable = 0x0 |
| 96 | + addr_pending = 0x1 |
| 97 | + |
| 98 | + def process(): |
| 99 | + yield sub.i.eq(1) |
| 100 | + yield Delay() |
| 101 | + self.assertEqual((yield sub.trg), 1) |
| 102 | + self.assertEqual((yield dut.src.i), 0) |
| 103 | + |
| 104 | + yield dut.bus.addr.eq(addr_enable) |
| 105 | + yield dut.bus.r_stb.eq(1) |
| 106 | + yield |
| 107 | + yield dut.bus.r_stb.eq(0) |
| 108 | + yield |
| 109 | + self.assertEqual((yield dut.bus.r_data), 0b0) |
| 110 | + yield |
| 111 | + |
| 112 | + yield dut.bus.addr.eq(addr_enable) |
| 113 | + yield dut.bus.w_stb.eq(1) |
| 114 | + yield dut.bus.w_data.eq(0b1) |
| 115 | + yield |
| 116 | + yield dut.bus.w_stb.eq(0) |
| 117 | + yield; yield Delay() |
| 118 | + |
| 119 | + self.assertEqual((yield dut.src.i), 1) |
| 120 | + yield sub.i.eq(0) |
| 121 | + yield Delay() |
| 122 | + self.assertEqual((yield sub.trg), 0) |
| 123 | + |
| 124 | + yield dut.bus.addr.eq(addr_pending) |
| 125 | + yield dut.bus.r_stb.eq(1) |
| 126 | + yield |
| 127 | + yield dut.bus.r_stb.eq(0) |
| 128 | + yield |
| 129 | + self.assertEqual((yield dut.bus.r_data), 0b1) |
| 130 | + yield |
| 131 | + |
| 132 | + yield dut.bus.addr.eq(addr_pending) |
| 133 | + yield dut.bus.w_stb.eq(1) |
| 134 | + yield dut.bus.w_data.eq(0b1) |
| 135 | + yield |
| 136 | + yield dut.bus.w_stb.eq(0) |
| 137 | + yield |
| 138 | + |
| 139 | + yield dut.bus.addr.eq(addr_pending) |
| 140 | + yield dut.bus.r_stb.eq(1) |
| 141 | + yield |
| 142 | + yield dut.bus.r_stb.eq(0) |
| 143 | + yield |
| 144 | + self.assertEqual((yield dut.bus.r_data), 0b0) |
| 145 | + |
| 146 | + simulation_test(dut, process) |
0 commit comments