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

nmigen.hdl.rec: restore Record.shape().

Browse source 
This method was lost in commit abbebf8e.
This commit is contained in:
awygle 2020-11-17 11:36:58 -08:00 committed by GitHub
parent 44318149e0
commit c1ed90807b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
nmigen/hdl/rec.py
View file

@ -198,6 +198,9 @@ class Record(ValueCastable):
name = "<unnamed>" name = "<unnamed>"
return "(rec {} {})".format(name, " ".join(fields)) return "(rec {} {})".format(name, " ".join(fields))
def shape(self):
return self.as_value().shape()
def connect(self, *subordinates, include=None, exclude=None): def connect(self, *subordinates, include=None, exclude=None):
def rec_name(record): def rec_name(record):
if record.name is None: if record.name is None:

4
tests/test_hdl_rec.py
View file

@ -176,6 +176,10 @@ class RecordTestCase(FHDLTestCase):
self.assertIs(r.stb, ns) self.assertIs(r.stb, ns)
self.assertIs(r.info, nr) self.assertIs(r.info, nr)
def test_shape(self):
r1 = Record([("a", 1), ("b", 2)])
self.assertEqual(r1.shape(), unsigned(3))
def test_like(self): def test_like(self):
r1 = Record([("a", 1), ("b", 2)]) r1 = Record([("a", 1), ("b", 2)])
r2 = Record.like(r1) r2 = Record.like(r1)