update ut

Author: wentaoyu

test/cpp/pir/distributed/dist_dialect_test.cc

@@ -165,8 +165,11 @@ TEST(operation_dist_attr_test, base) {
   EXPECT_EQ(op_attr, op_attr_1);
   EXPECT_EQ(op_attr.process_mesh(), process_mesh);
   EXPECT_EQ(op_attr.operand_dist_attrs(), operand_dist_attrs);
-  EXPECT_EQ(op_attr.result_dist_attrs(), result_dist_attrs);
   EXPECT_EQ(op_attr.operand_dist_attr(0), operand_dist_attrs.at(0));
   EXPECT_EQ(op_attr.operand_dist_attr(1), operand_dist_attrs.at(1));
+  EXPECT_EQ(op_attr.num_operand_dist_attrs(), (uint32_t)2);
+
+  EXPECT_EQ(op_attr.result_dist_attrs(), result_dist_attrs);
   EXPECT_EQ(op_attr.result_dist_attr(0), result_dist_attrs.at(0));
+  EXPECT_EQ(op_attr.num_result_dist_attrs(), (uint32_t)1);
 }