/
test_memblock.py
60 lines (50 loc) · 2.8 KB
/
test_memblock.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import unittest
import random
import pyrtl
from helperfunctions import *
# -------------------------------------------------------------------
class TestRTLMemBlockDesign(unittest.TestCase):
def setUp(self):
pyrtl.reset_working_block()
self.bitwidth = 3
self.addrwidth = 5
self.output1 = pyrtl.Output(self.bitwidth, "output1")
self.output2 = pyrtl.Output(self.bitwidth, "output2")
self.mem_read_address1 = pyrtl.Input(self.addrwidth, name='mem_read_address1')
self.mem_read_address2 = pyrtl.Input(self.addrwidth, name='mem_read_address2')
self.mem_write_address = pyrtl.Input(self.addrwidth, name='mem_write_address')
self.mem_write_data = pyrtl.Input(self.bitwidth, name='mem_write_data')
def tearDown(self):
pyrtl.reset_working_block()
def test_memblock_simple(self):
memory = pyrtl.MemBlock(bitwidth=self.bitwidth, addrwidth=self.addrwidth, name='memory')
self.output1 <<= memory[self.mem_read_address1]
self.output2 <<= memory[self.mem_read_address2]
memory[self.mem_write_address] <<= self.mem_write_data
def test_memblock_with_write_enable_with_equalsign(self):
memory = pyrtl.MemBlock(bitwidth=self.bitwidth, addrwidth=self.addrwidth, name='memory')
we = pyrtl.Const(1, bitwidth=1)
self.output1 <<= memory[self.mem_read_address1]
self.output2 <<= memory[self.mem_read_address2]
memory[self.mem_write_address] <<= pyrtl.MemBlock.EnabledWrite(self.mem_write_data, enable=we)
def test_memblock_direct_assignment_error(self):
memory = pyrtl.MemBlock(bitwidth=self.bitwidth, addrwidth=self.addrwidth, name='memory')
with self.assertRaises(pyrtl.PyrtlError):
memory[self.mem_write_address] = self.mem_write_data
def test_memblock_direct_assignment_error(self):
memory = pyrtl.MemBlock(bitwidth=self.bitwidth, addrwidth=self.addrwidth, name='memory')
memory[self.mem_write_address] <<= 5
""" -- TEST for Error comparing Mem reads #57
def test_memblock_to_memblock_direct_operation(self):
memory = pyrtl.MemBlock(bitwidth=self.bitwidth, addrwidth=self.addrwidth, name='memory')
temp = (memory[self.mem_read_address1] == memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] != memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] & memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] | memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] + memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] - memory[self.mem_read_address2])
temp = (memory[self.mem_read_address1] * memory[self.mem_read_address2])
self.output1 <<= temp
"""
if __name__ == "__main__":
unittest.main()