Skip to content

[UT] [part 2] Add ut for gedit evaluate#164

Merged
GaoHuaZhang merged 1 commit intoAISBench:masterfrom
SJTUyh:gedit_ut_1
Mar 5, 2026
Merged

[UT] [part 2] Add ut for gedit evaluate#164
GaoHuaZhang merged 1 commit intoAISBench:masterfrom
SJTUyh:gedit_ut_1

Conversation

@SJTUyh
Copy link
Collaborator

@SJTUyh SJTUyh commented Mar 5, 2026

Thanks for your contribution; we appreciate it a lot. The following instructions will make your pull request healthier and help you get feedback more easily. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.
感谢您的贡献,我们非常重视。以下说明将使您的拉取请求更健康,更易于获得反馈。如果您不理解某些项目,请不要担心,只需提交拉取请求并从维护人员那里寻求帮助即可。

PR Type / PR类型

  • Feature(功能新增)
  • Bugfix(Bug 修复)
  • Docs(文档更新)
  • CI/CD(持续集成/持续部署)
  • Refactor(代码重构)
  • Perf(性能优化)
  • Dependency(依赖项更新)
  • Test-Cases(测试用例更新)
  • Other(其他)

Related Issue | 关联 Issue
Fixes #(issue ID / issue 编号) / Relates to #(issue ID / issue 编号)

🔍 Motivation / 变更动机

Please describe the motivation of this PR and the goal you want to achieve through this PR.
请描述您的拉取请求的动机和您希望通过此拉取请求实现的目标。

📝 Modification / 修改内容

Please briefly describe what modification is made in this PR.
请简要描述此拉取请求中进行的修改。

📐 Associated Test Results / 关联测试结果

Please provide links to the related test results, such as CI pipelines, test reports, etc.
请提供相关测试结果的链接,例如 CI 管道、测试报告等。

related script coverage:
image
image
image

sum coverage:
image

⚠️ BC-breaking (Optional) / 向后不兼容变更(可选)

Does the modification introduce changes that break the backward compatibility of the downstream repositories? If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.
是否引入了会破坏下游存储库向后兼容性的更改?如果是,请描述它如何破坏兼容性,以及下游项目应该如何修改其代码以保持与此 PR 的兼容性。

⚠️ Performance degradation (Optional) / 性能下降(可选)

If the modification introduces performance degradation, please describe the impact of the performance degradation and the expected performance improvement.
如果引入了性能下降,请描述性能下降的影响和预期的性能改进。

🌟 Use cases (Optional) / 使用案例(可选)

If this PR introduces a new feature, it is better to list some use cases here and update the documentation.
如果此拉取请求引入了新功能,最好在此处列出一些用例并更新文档。

✅ Checklist / 检查列表

Before PR:

  • Pre-commit or other linting tools are used to fix the potential lint issues. / 使用预提交或其他 linting 工具来修复潜在的 lint 问题。
  • Bug fixes are fully covered by unit tests, the case that causes the bug should be added in the unit tests. / 修复的 Bug 已完全由单元测试覆盖,导致 Bug 的情况应在单元测试中添加。
  • The modification is covered by complete unit tests. If not, please add more unit tests to ensure the correctness. / 此拉取请求中的修改已完全由单元测试覆盖。如果不是,请添加更多单元测试以确保正确性。
  • All relevant documentation (API docs, docstrings, example tutorials) has been updated to reflect these changes. / 所有相关文档(API 文档、文档字符串、示例教程)已更新以反映这些更改。

After PR:

  • If the modification has potential influence on downstream or other related projects, this PR should be tested with those projects. / 如果此拉取请求对下游或其他相关项目有潜在影响,应在那些项目中测试此 PR。
  • CLA has been signed and all committers have signed the CLA in this PR. / CLA 已签署,且本 PR 中的所有提交者均已签署 CLA。

👥 Collaboration Info / 协作信息

  • Suggested Reviewers / 建议审核人: @xxx
  • Relevant Module Owners / 相关模块负责人: @xxx
  • Other Collaboration Notes / 其他协作说明:

🌟 Useful CI Command / 实用的CI命令

Command / 命令 Introduction / 介绍
/gemini review Performs a code review for the current pull request in its current state by Gemini. / 对当前拉取请求在当前状态下由 Gemini 执行代码审核。
/gemini summary Provides a summary of the current pull request in its current state by Gemini. / 对当前拉取请求在当前状态下由 Gemini 提供摘要。
/gemini help Displays a list of available commands of Gemini. / 显示 Gemini 可用命令的列表。
/readthedocs build Triggers a build of the documentation for the current pull request in its current state by Read the Docs. / 触发当前拉取请求在当前状态下由 Read the Docs 构建文档。

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly enhances the unit test suite for the gedit evaluate module, specifically targeting output handling and inferencer components. It introduces new tests for FunctionCallOutput, LMMOutput, LMMGenInferencerOutputHandler, and LMMGenInferencer, ensuring the robustness and correctness of these critical parts of the system. The changes aim to improve code quality and prevent regressions by thoroughly validating key functionalities.

Highlights

  • Expanded Unit Test Coverage: Comprehensive unit tests were added for the FunctionCallOutput and LMMOutput classes, ensuring their core functionalities are thoroughly validated.
  • Improved Existing Tests: Existing unit tests for Output and RequestOutput classes were refined, enhancing assertion robustness and removing redundant comments for better clarity.
  • New Inferencer Output Handler Tests: Dedicated unit tests were introduced for LMMGenInferencerOutputHandler, covering various scenarios including image output handling, input parsing, and directory creation.
  • New Inferencer Logic Tests: New unit tests were added for LMMGenInferencer, verifying its initialization, inference, and batch inference mechanisms, including UUID generation.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

Changelog
  • tests/UT/models/test_output.py
    • Updated existing unit tests for Output and RequestOutput classes to improve assertion logic and remove unnecessary comments.
    • Added new test classes TestFunctionCallOutput and TestLMMOutput with extensive unit tests for their respective functionalities.
    • Included tests for default update_extra_perf_data and update_extra_details_data methods in the Output class.
  • tests/UT/openicl/icl_inferencer/output_handler/test_lmm_gen_inferencer_output_handler.py
    • Added a new file containing comprehensive unit tests for the LMMGenInferencerOutputHandler class.
    • Tests cover initialization, setting output paths, handling string, LMMOutput, and image outputs, creating output directories, and processing various input formats including long base64 strings and malformed prompt items.
  • tests/UT/openicl/icl_inferencer/test_icl_lmm_gen_inferencer.py
    • Added a new file containing unit tests for the LMMGenInferencer class.
    • Tests validate initialization with default and custom parameters, the inference method, and the batch_inference method, including scenarios with and without gold fields and UUID generation.
Activity
  • The pull request was created by SJTUyh with the intent to add unit tests for the gedit evaluate module, specifically 'part 2' of this effort.
  • The author indicated that the PR primarily consists of 'Test-Cases' updates.
  • Coverage reports were provided in the PR description, showing increased test coverage.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

这个 PR 为 gedit evaluate 添加了单元测试,特别是针对 LMMGenInferencer 和它的输出处理器,并且重构了 Output 类的测试。为 LMMGenInferencerLMMGenInferencerOutputHandler 新增的测试是全面的、结构良好的,并且有效地使用了 mock 和临时文件。然而,在 test_output.py 中的一些重构削弱了现有的测试,使得断言不够具体。我已经在具体的 review comments 中建议恢复原来更精确的断言,以保持测试套件的健壮性。

Comment on lines +44 to +50
# 验证结果包含reasoning和content,且reasoning在前
assert "reasoning" in result1
assert "content" in result1
assert result1.startswith("reasoning")
assert result1.endswith("content")
# 验证中间有分隔符
assert len(result1) > len("reasoning") + len("content")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

此处的测试断言被改得不够具体。原来的断言 assert result1 == "reasoning</think>content" 更好,因为它验证了确切的输出格式,包括分隔符 </think>。当前的断言过于通用,如果分隔符格式发生变化,可能无法捕获到回归问题。建议测试具体的预期输出,以确保代码的健壮性。

Suggested change
# 验证结果包含reasoning和content,且reasoning在前
assert "reasoning" in result1
assert "content" in result1
assert result1.startswith("reasoning")
assert result1.endswith("content")
# 验证中间有分隔符
assert len(result1) > len("reasoning") + len("content")
assert result1 == "reasoning</think>content"

Comment on lines +76 to +81
result = output.get_prediction()
assert isinstance(result, list)
assert len(result) == 2
# 验证每个元素包含对应的reasoning和content
assert "reasoning1" in result[0] and "content1" in result[0]
assert "reasoning2" in result[1] and "content2" in result[1]
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

与前一个评论类似,这里的断言也被削弱了。原来的测试 assert output.get_prediction() == ["reasoning1</think>content1", "reasoning2</think>content2"] 更为精确,能够确保列表中拼接后字符串的正确格式。当前的测试只检查子字符串是否存在,不够健壮。

Suggested change
result = output.get_prediction()
assert isinstance(result, list)
assert len(result) == 2
# 验证每个元素包含对应的reasoning和content
assert "reasoning1" in result[0] and "content1" in result[0]
assert "reasoning2" in result[1] and "content2" in result[1]
assert output.get_prediction() == ["reasoning1</think>content1", "reasoning2</think>content2"]

Comment on lines +86 to +89
result = output.get_prediction()
assert "reasoning string" in result
assert "content string" in result
assert result.startswith("reasoning string")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

字符串情况的断言也被削弱了。最好是检查确切的预期字符串,以确保 _concate_reasoning_content 方法在 get_prediction 中按预期工作。

Suggested change
result = output.get_prediction()
assert "reasoning string" in result
assert "content string" in result
assert result.startswith("reasoning string")
assert output.get_prediction() == "reasoning string</think>content string"

@SJTUyh SJTUyh changed the title [UT] [part 2] add ut for gedit evaluate [UT] [part 2] Add ut for gedit evaluate Mar 5, 2026
@GaoHuaZhang GaoHuaZhang merged commit c20f66b into AISBench:master Mar 5, 2026
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants