()
| 21 | @pytest.mark.skipif(not torch.cuda.is_available(), reason="No CUDA device") |
| 22 | @pytest.fixture |
| 23 | def test_data(): |
| 24 | ( |
| 25 | means, |
| 26 | quats, |
| 27 | scales, |
| 28 | opacities, |
| 29 | colors, |
| 30 | viewmats, |
| 31 | Ks, |
| 32 | width, |
| 33 | height, |
| 34 | ) = load_test_data( |
| 35 | device=device, |
| 36 | data_path=os.path.join(os.path.dirname(__file__), "../assets/test_garden.npz"), |
| 37 | ) |
| 38 | colors = colors[None].repeat(len(viewmats), 1, 1) |
| 39 | return { |
| 40 | "means": means, |
| 41 | "quats": quats, |
| 42 | "scales": scales, |
| 43 | "opacities": opacities, |
| 44 | "colors": colors, |
| 45 | "viewmats": viewmats, |
| 46 | "Ks": Ks, |
| 47 | "width": width, |
| 48 | "height": height, |
| 49 | } |
| 50 | |
| 51 | |
| 52 | @pytest.mark.skipif(not torch.cuda.is_available(), reason="No CUDA device") |
nothing calls this directly
no test coverage detected