()
| 119 | |
| 120 | |
| 121 | def test_edge_case_zero_vector(): |
| 122 | engine = TurboQuantEngine(head_dim=128, total_bits=3, device="cpu") |
| 123 | K = torch.zeros(1, 128).half() |
| 124 | compressed = engine.compress_keys_pytorch(K) |
| 125 | assert not torch.isnan(compressed["k_mse"]).any() |
| 126 | assert not torch.isinf(compressed["k_mse"]).any() |
| 127 | |
| 128 | |
| 129 | def test_edge_case_large_values(): |
nothing calls this directly
no test coverage detected