fix

Author: YunchuWang

tests/test_eventgrid.py

@@ -14,10 +14,6 @@ class MyTestCase(unittest.TestCase):
     def test_eventgrid_input_type(self):
         check_input_type = azf_event_grid.EventGridEventInConverter.\
             check_input_type_annotation
-
-        if sys.version_info >= (3, 9):
-            self.assertTrue(check_input_type(list[func.EventGridEvent]))
-
         self.assertTrue(check_input_type(func.EventGridEvent))
         self.assertFalse(check_input_type(List[func.EventGridEvent]))
         self.assertFalse(check_input_type(str))
@@ -28,7 +24,7 @@ def test_eventgrid_output_type(self):
             check_output_type_annotation
 
         if sys.version_info >= (3, 9):
-            self.assertTrue(check_output_type(list[func.EventGridEvent]))
+            self.assertTrue(check_output_type(list[func.EventGridOutputEvent]))
             self.assertTrue(check_output_type(list[str]))
 
         self.assertTrue(check_output_type(func.EventGridOutputEvent))

tests/test_kafka.py

@@ -37,7 +37,6 @@ def test_kafka_input_type(self):
 
         if sys.version_info >= (3, 9):
             self.assertTrue(check_input_type(list[func.KafkaEvent]))
-            self.assertTrue(check_input_type(list[str]))
 
         self.assertTrue(check_input_type(func.KafkaEvent))
         self.assertTrue(check_input_type(List[func.KafkaEvent]))
@@ -51,9 +50,7 @@ def test_kafka_output_type(self):
         )
 
         if sys.version_info >= (3, 9):
-            self.assertTrue(check_output_type(list[func.KafkaEvent]))
             self.assertTrue(check_output_type(list[str]))
-            self.assertTrue(check_output_type(list[bytes]))
 
         self.assertTrue(check_output_type(bytes))
         self.assertTrue(check_output_type(str))