test: Add basic python tests for dicts/lists

Author: taminob

test/python_tests/python_tests.cpp

@@ -38,3 +38,39 @@ TEST_F(PythonTest, callFunctionWithWrongArguments)
     EXPECT_FALSE(result_1.valueOr(true));
     EXPECT_FALSE(result_2.valueOr(true));
 }
+
+TEST_F(PythonTest, callFunctionWithDict)
+{
+    auto result = plugin->call<int>("accept_dict", std::map<std::string, int> {
+                                                       { "", 1 },
+                                                       { "a", 2 },
+                                                       { "b", 3 },
+                                                       { "c", 4 },
+                                                   },
+        "c");
+
+    ASSERT_TRUE(result.hasValue()) << ppplugin::test::errorOutput(result);
+    EXPECT_EQ(result.valueOr(0), 4);
+}
+
+TEST_F(PythonTest, callFunctionWithList)
+{
+    auto result = plugin->call<std::string>("accept_list", std::vector<char> { 'a', 'b', 'c' });
+
+    ASSERT_TRUE(result.hasValue()) << ppplugin::test::errorOutput(result);
+    EXPECT_EQ(result.valueOr(""), "a,b,c,");
+}
+
+TEST_F(PythonTest, nestedDict)
+{
+    using ResultType = std::map<std::string, std::vector<std::map<int, int>>>;
+    const ResultType expected = {
+        { "", {} },
+        { "a", { { { 1, 1 }, { 2, 2 } }, { { 0, 0 } } } },
+        { "b", { { { 1, 1 } } } }
+    };
+
+    auto result = plugin->call<ResultType>("identity", expected);
+
+    EXPECT_EQ(result.valueOr(ResultType {}), expected);
+}

test/python_tests/test.py

@@ -7,5 +7,15 @@ def accept_int_string_bool_float(i, s, b, f):
     return is_int and is_string and is_bool and is_float
 
 
+def accept_dict(d, key):
+    return d[key]
+
+def accept_list(l):
+    result = ""
+    for e in l:
+        result += e + ","
+    return result
+
+
 def identity(x):
     return x