add testcase for code_block data_reflow

LazyAGI · Nov 12, 2024 · 98c28e3 · 98c28e3
1 parent 18e443c
commit 98c28e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/basic_tests/test_engine.py b/tests/basic_tests/test_engine.py
@@ -71,7 +71,8 @@ def test_engine_code(self):
     def test_engine_switch(self):
         plus1 = dict(id='1', kind='Code', name='m1', args=dict(code='def test(x: int):\n    return 1 + x\n'))
         double = dict(id='2', kind='Code', name='m2', args=dict(code='def test(x: int):\n    return 2 * x\n'))
-        square = dict(id='3', kind='Code', name='m3', args=dict(code='def test(x: int):\n    return x * x\n'))
+        square = dict(id='3', kind='Code', name='m3',
+                      args=dict(code='def test(x: int):\n    return x * x\n', _lazyllm_enable_report=True))
         switch = dict(
             id="4",
             kind="Switch",