Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: Fix printing of generic function parameters #516

Merged
merged 2 commits into from
Oct 1, 2024
Merged

fix: Fix printing of generic function parameters #516

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5f5662c
fix: Fix printing of generic function parameters
mark-koch Oct 1, 2024
aa68b93
Fix test
mark-koch Oct 1, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions guppylang/tys/param.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ def to_existential(self) -> tuple[Argument, ExistentialVar]:
var = ExistentialTypeVar.fresh(self.name, self.can_be_linear)
return TypeArg(var), var

def to_bound(self, idx: int | None = None) -> Argument:
def to_bound(self, idx: int | None = None) -> TypeArg:
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Drive by: Use a more precise return type which is used in the test below

"""Creates a bound variable with a given index that can be instantiated for this
parameter.
"""
Expand Down Expand Up @@ -169,7 +169,7 @@ def to_existential(self) -> tuple[Argument, ExistentialVar]:
var = ExistentialConstVar.fresh(self.name, self.ty)
return ConstArg(var), var

def to_bound(self, idx: int | None = None) -> Argument:
def to_bound(self, idx: int | None = None) -> ConstArg:
"""Creates a bound variable with a given index that can be instantiated for this
parameter.
"""
Expand Down
4 changes: 2 additions & 2 deletions guppylang/tys/printing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -128,12 +128,12 @@ def _visit_NumericType(self, ty: NumericType, inside_row: bool) -> str:
@_visit.register
def _visit_TypeParam(self, param: TypeParam, inside_row: bool) -> str:
# TODO: Print linearity?
return self.bound_names[-param.idx - 1]
return self.bound_names[param.idx]

@_visit.register
def _visit_ConstParam(self, param: ConstParam, inside_row: bool) -> str:
kind = self._visit(param.ty, True)
name = self.bound_names[-param.idx - 1]
name = self.bound_names[param.idx]
return f"{name}: {kind}"

@_visit.register
Expand Down
21 changes: 21 additions & 0 deletions tests/test_type_printing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
from guppylang.tys.builtin import array_type_def
from guppylang.tys.param import ConstParam, TypeParam
from guppylang.tys.ty import (
FunctionType,
FuncInput,
NumericType,
OpaqueType,
InputFlags,
)


def test_generic_function_type():
ty_param = TypeParam(0, "T", can_be_linear=True)
len_param = ConstParam(1, "n", NumericType(NumericType.Kind.Nat))
array_ty = OpaqueType([ty_param.to_bound(0), len_param.to_bound(1)], array_type_def)
ty = FunctionType(
params=[ty_param, len_param],
inputs=[FuncInput(array_ty, InputFlags.Inout)],
output=ty_param.to_bound(0).ty,
)
assert str(ty) == "forall T, n: nat. array[T, n] -> T"
Loading