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

hdl.ast: implement Value.__pos__.

Browse source
This commit is contained in:
Catherine 2023-02-27 22:31:17 +00:00
parent 342fd5f8bd
commit bef2052c1e
2 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
amaranth/hdl/ast.py
View file

@ -166,6 +166,9 @@ class Value(metaclass=ABCMeta):
def __bool__(self): def __bool__(self):
raise TypeError("Attempted to convert Amaranth value to Python boolean") raise TypeError("Attempted to convert Amaranth value to Python boolean")
def __pos__(self):
return self
def __invert__(self): def __invert__(self):
return Operator("~", [self]) return Operator("~", [self])
def __neg__(self): def __neg__(self):

3
tests/test_hdl_ast.py
View file

@ -397,6 +397,9 @@ class OperatorTestCase(FHDLTestCase):
self.assertEqual(repr(v), "(s (const 4'd1))") self.assertEqual(repr(v), "(s (const 4'd1))")
self.assertEqual(v.shape(), signed(4)) self.assertEqual(v.shape(), signed(4))
def test_pos(self):
self.assertRepr(+Const(10), "(const 4'd10)")
def test_neg(self): def test_neg(self):
v1 = -Const(0, unsigned(4)) v1 = -Const(0, unsigned(4))
self.assertEqual(repr(v1), "(- (const 4'd0))") self.assertEqual(repr(v1), "(- (const 4'd0))")