Add additional test case

Author: Wheest

mlir/test/python/dialects/func.py

@@ -109,11 +109,27 @@ def testFunctionCalls():
 # CHECK-LABEL: TEST: testFunctionArgAttrs
 @constructAndPrintInModule
 def testFunctionArgAttrs():
-    foo = func.FuncOp("foo", ([("arg0", F32Type.get())], []))
+    foo = func.FuncOp("foo", ([F32Type.get()], []))
+    foo.sym_visibility = StringAttr.get("private")
+    foo2 = func.FuncOp("foo2", ([F32Type.get(), F32Type.get()], []))
+    foo2.sym_visibility = StringAttr.get("private")
 
-    assert len(foo.arg_attrs) == 1
-    assert foo.arg_attrs[0] == DictAttr.get({})
+    empty_attr = DictAttr.get({})
+    test_attr = DictAttr.get({"test.foo": StringAttr.get("bar")})
+    test_attr2 = DictAttr.get({"test.baz": StringAttr.get("qux")})
 
-    foo.arg_attrs = [DictAttr.get({"test.foo": StringAttr.get("bar")})]
+    assert len(foo.arg_attrs) == 1
+    assert foo.arg_attrs[0] == empty_attr
 
+    foo.arg_attrs = [test_attr]
     assert foo.arg_attrs[0]["test.foo"] == StringAttr.get("bar")
+
+    assert len(foo2.arg_attrs) == 2
+    assert foo2.arg_attrs == ArrayAttr.get([empty_attr, empty_attr])
+
+    foo2.arg_attrs = [empty_attr, test_attr2]
+    assert foo2.arg_attrs == ArrayAttr.get([empty_attr, test_attr2])
+
+
+# CHECK: func private @foo(f32 {test.foo = "bar"})
+# CHECK: func private @foo2(f32, f32  {test.baz = "qux"})