()
| 108 | |
| 109 | |
| 110 | def test_compression_ratios(): |
| 111 | d = 128 |
| 112 | engine_3bit = TurboQuantEngine(head_dim=d, total_bits=3, device="cpu") |
| 113 | engine_4bit = TurboQuantEngine(head_dim=d, total_bits=4, device="cpu") |
| 114 | |
| 115 | stats_3 = engine_3bit.compressed_size_bytes(4096) |
| 116 | stats_4 = engine_4bit.compressed_size_bytes(4096) |
| 117 | |
| 118 | assert 4.5 < stats_3["compression_ratio"] < 6.0, ( |
| 119 | f"3-bit compression: {stats_3['compression_ratio']:.2f}x (expected ~5x)" |
| 120 | ) |
| 121 | assert 3.0 < stats_4["compression_ratio"] < 4.5, ( |
| 122 | f"4-bit compression: {stats_4['compression_ratio']:.2f}x (expected ~3.8x)" |
| 123 | ) |
| 124 | |
| 125 | |
| 126 | def test_deterministic_with_same_seed(): |
nothing calls this directly
no test coverage detected