Skip to content

Add unit tests for SamplingParams and CompilationConfig #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
ChuanLi1101 wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add unit tests for SamplingParams and CompilationConfig #32

ChuanLi1101 wants to merge 2 commits into from

Conversation

@ChuanLi1101
Copy link
Collaborator

@ChuanLi1101 ChuanLi1101 commented Dec 9, 2025 •
edited
Loading

Summary

This PR adds a basic unit test framework to the ATOM project.

Changes

  • Added tests/ directory with pytest-based unit tests
  • Added tests for SamplingParams dataclass
  • Added tests for CompilationConfig and CUDAGraphMode
  • Updated pyproject.toml with pytest dev dependencies and configuration

Test Coverage

  • test_sampling_params.py: Tests default values, custom values, and equality
  • test_config.py: Tests CUDAGraphMode enum methods and CompilationConfig validation

How to Run

pip install pytest
pytest tests/ -v## Notes

  • Tests are pure Python and do not require GPU hardware
  • Compatible with both NVIDIA CUDA and AMD ROCm environments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@carlushuang carlushuang Awaiting requested review from carlushuang

@valarLip valarLip Awaiting requested review from valarLip

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@ChuanLi1101