Test caching a kernel.
(self, cache_manager, sample_kernel)
| 181 | assert kernel_id != kernel_id3 |
| 182 | |
| 183 | def test_cache_kernel(self, cache_manager, sample_kernel): |
| 184 | """Test caching a kernel.""" |
| 185 | kernel_id = cache_manager.cache_kernel("llama-7b", "matmul", sample_kernel) |
| 186 | |
| 187 | assert isinstance(kernel_id, str) |
| 188 | |
| 189 | # Verify kernel file was created |
| 190 | kernel_file = cache_manager.kernels_dir / f"{kernel_id}.pkl" |
| 191 | assert kernel_file.exists() |
| 192 | |
| 193 | # Verify database entry |
| 194 | with sqlite3.connect(cache_manager.db_path) as conn: |
| 195 | cursor = conn.cursor() |
| 196 | cursor.execute("SELECT * FROM kernel_metadata WHERE id = ?", (kernel_id,)) |
| 197 | row = cursor.fetchone() |
| 198 | assert row is not None |
| 199 | |
| 200 | def test_get_cached_kernel_exists(self, cache_manager, sample_kernel): |
| 201 | """Test retrieving a cached kernel.""" |
nothing calls this directly
no test coverage detected