From f08a773111a8c888679efd92efc243ace8de213c Mon Sep 17 00:00:00 2001 From: Marek Materzok Date: Tue, 24 Oct 2023 09:53:17 +0200 Subject: [PATCH] Un-string some types. --- test/common/functions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/common/functions.py b/test/common/functions.py index 6a11e41ab..c4ffc814a 100644 --- a/test/common/functions.py +++ b/test/common/functions.py @@ -8,7 +8,7 @@ TestGen: TypeAlias = Generator[Command | Value | Statement | None, Any, T] -def set_inputs(values: RecordValueDict, field: Record) -> "TestGen[None]": +def set_inputs(values: RecordValueDict, field: Record) -> TestGen[None]: for name, value in values.items(): if isinstance(value, dict): yield from set_inputs(value, getattr(field, name)) @@ -16,7 +16,7 @@ def set_inputs(values: RecordValueDict, field: Record) -> "TestGen[None]": yield getattr(field, name).eq(value) -def get_outputs(field: Record) -> "TestGen[RecordIntDict]": +def get_outputs(field: Record) -> TestGen[RecordIntDict]: # return dict of all signal values in a record because amaranth's simulator can't read all # values of a Record in a single yield - it can only read Values (Signals) result = {}