| 1 |
"""Pure merge helper coverage.""" |
| 2 |
|
| 3 |
from __future__ import annotations |
| 4 |
|
| 5 |
import pytest |
| 6 |
|
| 7 |
from dlm.export.errors import UnsafeMergeError |
| 8 |
from dlm.export.merge import check_merge_safety |
| 9 |
from dlm.export.plan import ExportPlan |
| 10 |
|
| 11 |
|
| 12 |
def test_check_merge_safety_delegates_to_plan() -> None: |
| 13 |
check_merge_safety(ExportPlan(merged=False), was_qlora=True) |
| 14 |
|
| 15 |
|
| 16 |
def test_check_merge_safety_refuses_unsafe_qlora_merge() -> None: |
| 17 |
with pytest.raises(UnsafeMergeError, match="QLoRA"): |
| 18 |
check_merge_safety( |
| 19 |
ExportPlan(merged=True, dequantize_confirmed=False), |
| 20 |
was_qlora=True, |
| 21 |
) |