---
license: cc-by-4.0
base_model:
- Salesforce/codet5-large
---

# AsserT5: Test Assertion Generation Using a Fine-Tuned Code Language Model

Part of the replication package for our paper at AST 2025 (preprint: https://arxiv.org/abs/2502.02708).
To be used in combination with our main replication package at https://doi.org/10.5281/zenodo.14703162.

AsserT5 is a fine-tuned [CodeT5](https://huggingface.co/Salesfoce/codet5-large) trained to generate assertion statements for Java JUnit test cases.
It was trained on an extended variant of the [methods2test](https://github.com/microsoft/methods2test) dataset.


## Structure

- The top-level number indicates the maximum number of assertions allowed per test case in the training dataet.
- The next level below indicates the model variant:
    - `abstract`: Identifiers in the data are replaced with abstract tokens.
    - `raw`: The source code is tokenised as-is.
    - `test-method`: The model is trained only on the test case code rather than test case + focal method pairs.