From 766dc74a04bfd8f789b42d892d71ea33cc4c5a42 Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Sat, 27 Jan 2024 19:08:18 +0800
Subject: :white_check_mark: test(refact): sync tests with infini changes

---
 tests/test_core.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'tests/test_core.py')

diff --git a/tests/test_core.py b/tests/test_core.py
index db402dea..01d7ecb4 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -1,5 +1,5 @@
 from infini.core import Core
-from infini.generator import Generator
+from infini.generator import TextGenerator
 from infini.handler import Handler
 from infini.input import Input
 from infini.interceptor import Interceptor
@@ -48,7 +48,7 @@ def test_core():
         },
     ]
 
-    generator = Generator()
+    generator = TextGenerator()
     generator.events = {
         "test.cmd": "cmd",
         "test.add": "{{ result }}",
-- 
cgit v1.2.3-70-g09d2