Add tests for shiny_mode comment detection

posit-dev · Mar 20, 2024 · 2f75a90 · 2f75a90
1 parent 00d956e
commit 2f75a90
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/pytest/test_express.py b/tests/pytest/test_express.py
@@ -44,3 +44,10 @@ def write_tmp_file(s: str):
 
     write_tmp_file("def f():\n  import shiny.express")
     assert not express.is_express_app(tmp_file, None)
+
+    # Look for magic comment - should override import detection
+    write_tmp_file("\n#shiny_mode: core\nfrom shiny.express import ui")
+    assert not express.is_express_app(tmp_file, None)
+
+    write_tmp_file("#shiny_mode: express\nfrom shiny import ui")
+    assert express.is_express_app(tmp_file, None)