test: unreachable warning rendering tests

Author: ss2165

guppylang-internals/src/guppylang_internals/cfg/builder.py

@@ -26,7 +26,7 @@
 )
 from guppylang_internals.checker.errors.type_errors import WrongNumberOfArgsError
 from guppylang_internals.diagnostic import Error, Warning
-from guppylang_internals.error import GuppyError, InternalGuppyError, emit_warning
+from guppylang_internals.error import GuppyError, InternalGuppyError
 from guppylang_internals.experimental import (
     check_lists_enabled,
     check_modifiers_enabled,
@@ -48,6 +48,7 @@
 )
 from guppylang_internals.span import Span, to_span
 from guppylang_internals.tys.ty import NoneType, UnitaryFlags
+from guppylang_internals.warning import emit_warning
 
 # In order to build expressions, need an endless stream of unique temporary variables
 # to store intermediate results

tests/diagnostics/snapshots/miette/test_unreachable_warning.txt

@@ -0,0 +1,7 @@
+  ! Unreachable
+   ,-[3:5]
+ 2 |     return 0
+ 3 |     x = 1
+   :     ^^|^^
+   :       `-- This code is not reachable
+   `----

tests/diagnostics/snapshots/test_unreachable_warning.txt

@@ -0,0 +1,6 @@
+Warning: Unreachable (at <unknown>:3:4)
+  | 
+1 | def foo():
+2 |     return 0
+3 |     x = 1
+  |     ^^^^^ This code is not reachable

tests/diagnostics/test_diagnostics_rendering.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 from typing import ClassVar
 
+from guppylang_internals.cfg.builder import UnreachableWarning
 from guppylang_internals.diagnostic import (
     Diagnostic,
     DiagnosticsRenderer,
@@ -164,6 +165,13 @@ class MySubDiagnostic(Note):
     run_test(source, diagnostic, snapshot, request)
 
 
+def test_unreachable_warning(snapshot, request):
+    source = "def foo():\n    return 0\n    x = 1\n"
+    span = Span(Loc(file, 3, 4), Loc(file, 3, 9))
+    diagnostic = UnreachableWarning(span)
+    run_test(source, diagnostic, snapshot, request)
+
+
 def test_context(snapshot, request):
     source = "super_apple := apple ** 2\nlemon := orange - apple\napple == orange"
     span = Span(Loc(file, 3, 6), Loc(file, 3, 8))

tests/diagnostics/test_miette_rendering.py

@@ -5,6 +5,7 @@
 from typing import ClassVar
 
 import pytest
+from guppylang_internals.cfg.builder import UnreachableWarning
 from guppylang_internals.diagnostic import (
     Diagnostic,
     DiagnosticLevel,
@@ -107,6 +108,15 @@ class WarningDiag(Error):
     run_miette_test(source, diagnostic, snapshot, request)
 
 
+def test_unreachable_warning(snapshot, request):
+    """Test rendering of the concrete unreachable-code warning."""
+
+    source = "def foo():\n    return 0\n    x = 1\n"
+    span = Span(Loc(file, 3, 4), Loc(file, 3, 9))
+    diagnostic = UnreachableWarning(span)
+    run_miette_test(source, diagnostic, snapshot, request)
+
+
 def test_complete_issue_example(snapshot, request):
     """Test complete example from issue #968 with primary + sub-diagnostics."""
 

tests/error/test_misc_errors.py

@@ -1,7 +1,8 @@
 import pathlib
-import pytest
 
+import pytest
 from guppylang import guppy
+
 from tests.error.util import run_error_test
 
 path = pathlib.Path(__file__).parent.resolve() / "misc_errors"

tests/error/util.py

@@ -37,6 +37,7 @@ def filter_traceback_not_containing(s: str, disallowed_regex: re.Pattern[str]) -
 
     return "\n".join(result)
 
+
 def run_error_test(file, capsys, snapshot):
     file = pathlib.Path(file)
 

tests/integration/test_unreachable.py

@@ -6,7 +6,7 @@
 from guppylang import GuppyWarning
 from guppylang_internals.error import GuppyError
 from guppylang.std.quantum import discard, h
-from tests.util import compile_guppy
+from tests.util import compile_guppy, guppy_warning_records
 
 
 def assert_unreachable_warning_emitted(fn):
@@ -16,8 +16,9 @@ def assert_unreachable_warning_emitted(fn):
         warnings.simplefilter("always")
         result = fn()
 
-    assert len(records) == 1
-    warning = records[0]
+    guppy_records = guppy_warning_records(records)
+    assert len(guppy_records) == 1
+    warning = guppy_records[0]
     assert warning.category is GuppyWarning
     assert str(warning.message) == "Unreachable: This code is not reachable"
     return result
@@ -166,8 +167,9 @@ def test() -> int:
                 return 1
             return 0
 
-    assert len(records) == 1
-    warning = records[0]
+    guppy_records = guppy_warning_records(records)
+    assert len(guppy_records) == 1
+    warning = guppy_records[0]
     assert warning.category is GuppyWarning
     assert warning.filename.endswith("test_unreachable.py")
     assert str(warning.message) == "Unreachable: This code is not reachable"
@@ -185,8 +187,9 @@ def test() -> int:
             x = 1
             return x
 
-    assert len(records) == 1
-    warning = records[0]
+    guppy_records = guppy_warning_records(records)
+    assert len(guppy_records) == 1
+    warning = guppy_records[0]
     assert warning.category is GuppyWarning
     assert warning.filename.endswith("test_unreachable.py")
     assert str(warning.message) == "Unreachable: This code is not reachable"
@@ -205,4 +208,5 @@ def test() -> int:
                     return 1.0
                 return 0
 
-    assert len(records) == 0
+    guppy_records = guppy_warning_records(records)
+    assert len(guppy_records) == 0

tests/integration/test_warning_rendering.py

@@ -0,0 +1,28 @@
+import importlib
+import pathlib
+import warnings
+
+from tests.util import guppy_warning_records
+
+
+def run_warning_test(file: pathlib.Path, capsys, snapshot) -> None:
+    """Snapshot rich warning rendering for a module-level compiler warning."""
+    with warnings.catch_warnings(record=True) as records:
+        warnings.simplefilter("always")
+        importlib.import_module(f"tests.integration.{file.parent.name}.{file.stem}")
+
+    guppy_records = guppy_warning_records(records)
+    assert len(guppy_records) == 1
+    err = capsys.readouterr().err
+    err = err.replace(str(file), "$FILE")
+
+    snapshot.snapshot_dir = str(file.parent)
+    snapshot.assert_match(err, file.with_suffix(".err").name)
+
+
+def test_check_warning(capsys, snapshot):
+    """Rich warnings should snapshot the rendered diagnostic output."""
+    file = (
+        pathlib.Path(__file__).parent.resolve() / "warning_cases" / "check_warning.py"
+    )
+    run_warning_test(file, capsys, snapshot)

tests/integration/warning_cases/__init__.py

@@ -0,0 +1 @@
+