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_generator.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tests/test_generator.py')

diff --git a/tests/test_generator.py b/tests/test_generator.py
index bcab31ed..0eed5756 100644
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -1,9 +1,9 @@
-from infini.generator import Generator
+from infini.generator import TextGenerator
 from infini.output import Output
 
 
 def test_generator():
-    generator = Generator()
+    generator = TextGenerator()
     generator.events = {
         "test.event1": "Event1 文本",
     }
@@ -12,7 +12,7 @@ def test_generator():
 
 
 def test_generator_with_var():
-    generator = Generator()
+    generator = TextGenerator()
     generator.events = {
         "test.event1": "Event1 文本: {{ var }}",
     }
@@ -27,7 +27,7 @@ def test_generator_with_function():
     def add(a, b):
         return a + b
 
-    generator = Generator()
+    generator = TextGenerator()
     generator.events = {
         "test.event": "{{ func(1, 2) }}",
     }
-- 
cgit v1.2.3-70-g09d2