Skip to content

Tests for rest of the tool functions - #44

Merged
pxlxingliang merged 9 commits into
deepmodeling:developfrom
ahxbcn:develop
Sep 2, 2025
Merged

Tests for rest of the tool functions#44
pxlxingliang merged 9 commits into
deepmodeling:developfrom
ahxbcn:develop

Conversation

@ahxbcn

@ahxbcn ahxbcn commented Sep 1, 2025

Copy link
Copy Markdown
Contributor

No description provided.


return {'band_gap': plot_output['band_gap'],
'band_output_dir': Path(work_path).absolute(),
'band_calc_dir': Path(work_path).absolute(),

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I consider that we should highlight the change of input and output for any @mcp.tool function

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

I agree with you.
In this pull request, two modifications about return value were made . This modification in abacus_cal_band is for aligning keys in returned dictionary in pyatb and nscf mode.
Another modification is in abacus_vibration_analysis, which merges two return keys, real_frequencies and imaginary_frequencies, into one key, where imaginary frequencies is represented by real negative values now.
These modifications, along with tests, were made for improving stability and reducing bugs in @mcp.tool functions. Upgrade to v0.2 is expected for later bemchmarks on different platforms.

md_tfirst = 300)

self.assertIsInstance(outputs['md_work_path'], get_path_type())
self.assertIsInstance(outputs['md_traj_file'], get_path_type())

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Please add more the correctness checking of the results, such as normal_end, steps number.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Normal_end and number of frames contained in the trajectory to the return value of abacus_run_md and corresponding tests are added.

…eturned traj file in the return value of tool function abacus_run_md
@pxlxingliang
pxlxingliang merged commit d972a6f into deepmodeling:develop Sep 2, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants