Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
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 returns/primitives/tracing.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
import types
from contextlib import contextmanager
from inspect import FrameInfo, stack
from typing import List, Optional
from typing import Iterator, List, Optional

from returns.result import _Failure


@contextmanager
def collect_traces():
def collect_traces() -> Iterator[None]:
"""
Context Manager/Decorator to active traces collect to the Failures.

Expand Down
13 changes: 13 additions & 0 deletions typesafety/test_primitives/test_tracing/test_collect_traces.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
- case: collect_traces_context_manager_return_type_hint
disable_cache: true
main: |
from returns.primitives.tracing import collect_traces
reveal_type(collect_traces) # N: Revealed type is 'def () -> contextlib._GeneratorContextManager[None]'
- case: collect_traces_context_manager_return_type
disable_cache: true
main: |
from returns.primitives.tracing import collect_traces
reveal_type(collect_traces()) # N: Revealed type is 'contextlib._GeneratorContextManager[None]'