Add some more complex test cases

Author: autophagy

Diff for: flink-python/pyflink/table/tests/test_table_environment_api.py

@@ -700,19 +700,28 @@ def test_collect_for_all_data_types(self):
                                1.98932, bytearray(b'pyflink'), 'pyflink',
                                datetime.date(2014, 9, 13), datetime.time(12, 0, 0, 123000),
                                datetime.datetime(2018, 3, 11, 3, 0, 0, 123000),
+                               [['a', 'b'], ['c', 'd'], ['e', 'f']],
                                [Row('pyflink'), Row('pyflink'), Row('pyflink')],
                                {1: Row('flink'), 2: Row('pyflink')},
+                               [Row('a1', {1: Row('b1')}, [Row('c1', 'd1'), Row('e1', 'f1')]),
+                                Row('a2', {2: Row('b2')}, [Row('c2', 'd2'), Row('e2', 'f2')])],
                                decimal.Decimal('1000000000000000000.050000000000000000'),
                                decimal.Decimal('1000000000000000000.059999999999999999'))]
         source = self.t_env.from_elements(
             [(1, None, 1, True, 32767, -2147483648, 1.23, 1.98932, bytearray(b'pyflink'), 'pyflink',
               datetime.date(2014, 9, 13), datetime.time(hour=12, minute=0, second=0,
                                                         microsecond=123000),
               datetime.datetime(2018, 3, 11, 3, 0, 0, 123000),
+              [['a', 'b'], ['c', 'd'], ['e', 'f']],
               [Row('pyflink'), Row('pyflink'), Row('pyflink')],
-              {1: Row('flink'), 2: Row('pyflink')}, decimal.Decimal('1000000000000000000.05'),
-              decimal.Decimal('1000000000000000000.05999999999999999899999999999'))], DataTypes.ROW(
-                [DataTypes.FIELD("a", DataTypes.BIGINT()), DataTypes.FIELD("b", DataTypes.BIGINT()),
+              {1: Row('flink'), 2: Row('pyflink')},
+              [Row('a1', {1: Row('b1')}, [Row('c1', 'd1'), Row('e1', 'f1')]),
+               Row('a2', {2: Row('b2')}, [Row('c2', 'd2'), Row('e2', 'f2')])],
+              decimal.Decimal('1000000000000000000.05'),
+              decimal.Decimal('1000000000000000000.05999999999999999899999999999'))],
+            DataTypes.ROW(
+                [DataTypes.FIELD("a", DataTypes.BIGINT()),
+                 DataTypes.FIELD("b", DataTypes.BIGINT()),
                  DataTypes.FIELD("c", DataTypes.TINYINT()),
                  DataTypes.FIELD("d", DataTypes.BOOLEAN()),
                  DataTypes.FIELD("e", DataTypes.SMALLINT()),
@@ -724,12 +733,52 @@ def test_collect_for_all_data_types(self):
                  DataTypes.FIELD("k", DataTypes.DATE()),
                  DataTypes.FIELD("l", DataTypes.TIME()),
                  DataTypes.FIELD("m", DataTypes.TIMESTAMP(3)),
-                 DataTypes.FIELD("n", DataTypes.ARRAY(DataTypes.ROW([DataTypes.FIELD('ss2',
-                                                                     DataTypes.STRING())]))),
-                 DataTypes.FIELD("o", DataTypes.MAP(DataTypes.BIGINT(), DataTypes.ROW(
-                     [DataTypes.FIELD('ss', DataTypes.STRING())]))),
-                 DataTypes.FIELD("p", DataTypes.DECIMAL(38, 18)), DataTypes.FIELD("q",
-                 DataTypes.DECIMAL(38, 18))]))
+                 DataTypes.FIELD("n", DataTypes.ARRAY(DataTypes.ARRAY(DataTypes.STRING()))),
+                 DataTypes.FIELD(
+                     "o",
+                     DataTypes.ARRAY(
+                         DataTypes.ROW([DataTypes.FIELD("ss2", DataTypes.STRING())])
+                     )),
+                 DataTypes.FIELD(
+                     "p",
+                     DataTypes.MAP(
+                         DataTypes.BIGINT(),
+                         DataTypes.ROW([DataTypes.FIELD("ss", DataTypes.STRING())]),
+                     )),
+                 DataTypes.FIELD(
+                     "q",
+                     DataTypes.ARRAY(
+                         DataTypes.ROW(
+                             [
+                                 DataTypes.FIELD("a1", DataTypes.STRING()),
+                                 DataTypes.FIELD(
+                                     "a2",
+                                     DataTypes.MAP(
+                                         DataTypes.BIGINT(),
+                                         DataTypes.ROW(
+                                             [DataTypes.FIELD("ss", DataTypes.STRING())]
+                                         ),
+                                     ),
+                                 ),
+                                 DataTypes.FIELD(
+                                     "a3",
+                                     DataTypes.ARRAY(
+                                         DataTypes.ROW(
+                                             [
+                                                 DataTypes.FIELD("a1", DataTypes.STRING()),
+                                                 DataTypes.FIELD("a2", DataTypes.STRING()),
+                                             ]
+                                         )
+                                     ),
+                                 ),
+                             ]
+                         )
+                     )),
+                 DataTypes.FIELD("r", DataTypes.DECIMAL(38, 18)),
+                 DataTypes.FIELD("s", DataTypes.DECIMAL(38, 18))
+                 ]
+            )
+        )
         table_result = source.execute()
         with table_result.collect() as result:
             collected_result = []