Refactor ListEncoderDecoderTests to use EvaluatePythonScript

Co-authored-by: ipetrov <[email protected]>

Author: cursoragent

DSpythonNet3Tests/ListEncoderDecoderTests.cs

@@ -1,8 +1,6 @@
 using System.Collections;
-using DSPythonNet3;
-using DSPythonNet3.Encoders;
 using NUnit.Framework;
-using Python.Runtime;
+using DSPythonNet3;
 
 namespace DSPythonNet3Tests
 {
@@ -13,35 +11,46 @@ public void TryDecode_ConvertsNestedPythonListsToClrLists()
         {
             DSPythonNet3Evaluator.InitializePython();
 
-            using (Py.GIL())
-            using (var scope = Py.CreateScope())
-            {
-                scope.Exec("value = [[1, [2, 3]], ['a', ['b']]]");
-                using var pyList = scope.Get("value");
-
-                var decoder = new ListEncoderDecoder();
-                var success = decoder.TryDecode(pyList, out IList result);
+            string code = @"
+import clr
+clr.AddReference('DSCoreNodes')
+from DSCore import List
 
-                Assert.That(success, Is.True);
-                Assert.That(result, Is.InstanceOf<IList>());
-
-                var first = result[0] as IList;
-                Assert.That(first, Is.Not.Null);
-                Assert.That(first[0], Is.EqualTo(1));
-
-                var secondLevel = first?[1] as IList;
-                Assert.That(secondLevel, Is.Not.Null);
-                Assert.That(secondLevel?[0], Is.EqualTo(2));
-                Assert.That(secondLevel?[1], Is.EqualTo(3));
+data = [[1, 2, 3], [4, 5, 6]]
+OUT = data, List.Flatten(data, -1)
+";
+            var empty = new ArrayList();
+            var expected = new ArrayList
+            {
+                new ArrayList
+                {
+                    new ArrayList { 1, 2, 3 },
+                    new ArrayList { 4, 5, 6 }
+                },
+                new ArrayList { 1, 2, 3, 4, 5, 6 }
+            };
+
+            var result = DSPythonNet3Evaluator.EvaluatePythonScript(code, empty, empty);
+            Assert.That(result, Is.InstanceOf<IEnumerable>());
+
+            var normalizedResult = NormalizeResult(result);
+            CollectionAssert.AreEqual(expected, normalizedResult as IEnumerable);
+        }
 
-                var second = result[1] as IList;
-                Assert.That(second, Is.Not.Null);
-                Assert.That(second?[0], Is.EqualTo("a"));
+        private static object NormalizeResult(object value)
+        {
+            if (value is string || value is not IEnumerable enumerable)
+            {
+                return value;
+            }
 
-                var thirdLevel = second?[1] as IList;
-                Assert.That(thirdLevel, Is.Not.Null);
-                Assert.That(thirdLevel?[0], Is.EqualTo("b"));
+            var list = new ArrayList();
+            foreach (var item in enumerable)
+            {
+                list.Add(NormalizeResult(item));
             }
+
+            return list;
         }
     }
 }