usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

hdl.rec: fix using Enum subclass as shape if direction is specified.

Browse source 
Also improves error messages.

Fixes #224.
This commit is contained in:
whitequark 2019-09-22 17:23:32 +00:00
parent 4c582ef609
commit 1976310bf0
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
nmigen/test/test_hdl_rec.py
View file

@ -36,8 +36,10 @@ class LayoutTestCase(FHDLTestCase):
def test_enum_field(self):
layout = Layout.wrap([
("enum", UnsignedEnum),
("enum_dir", UnsignedEnum, DIR_FANOUT),
])
self.assertEqual(layout["enum"], ((2, False), DIR_NONE))
self.assertEqual(layout["enum_dir"], ((2, False), DIR_FANOUT))
def test_slice_tuple(self):
layout = Layout.wrap([
@ -75,7 +77,7 @@ class LayoutTestCase(FHDLTestCase):
def test_wrong_shape(self):
with self.assertRaises(TypeError,
msg="Field ('a', 'x') has invalid shape: should be an int, tuple, or "
msg="Field ('a', 'x') has invalid shape: should be an int, tuple, Enum, or "
"list of fields of a nested record"):
Layout.wrap([("a", "x")])