|
4 | 4 | from pytest_container import DerivedContainer |
5 | 5 | from pytest_container import container_and_marks_from_pytest_param |
6 | 6 | from pytest_container.container import ContainerData |
7 | | -from pytest_container.runtime import LOCALHOST |
8 | 7 |
|
9 | 8 | from bci_tester.data import NVIDIA_CONTAINERS |
10 | 9 |
|
@@ -44,33 +43,19 @@ def test_image_content(container_per_test: ContainerData): |
44 | 43 | "/usr/local/bin/nvidia-driver-selector.sh", |
45 | 44 | ] |
46 | 45 |
|
47 | | - if LOCALHOST.system_info.arch in ("aarch64",): |
48 | | - ### TODO why are those uncompressed? |
49 | | - files += [ |
50 | | - "/opt/open/nvidia-drm.ko", |
51 | | - "/opt/open/nvidia-modeset.ko", |
52 | | - "/opt/open/nvidia-peermem.ko", |
53 | | - "/opt/open/nvidia-uvm.ko", |
54 | | - "/opt/open/nvidia.ko", |
55 | | - "/opt/proprietary/nvidia-drm.ko", |
56 | | - "/opt/proprietary/nvidia-modeset.ko", |
57 | | - "/opt/proprietary/nvidia-peermem.ko", |
58 | | - "/opt/proprietary/nvidia-uvm.ko", |
59 | | - "/opt/proprietary/nvidia.ko", |
60 | | - ] |
61 | | - elif LOCALHOST.system_info.arch in ("x86_64",): |
62 | | - files += [ |
63 | | - "/opt/open/nvidia-drm.ko.zst", |
64 | | - "/opt/open/nvidia-modeset.ko.zst", |
65 | | - "/opt/open/nvidia-peermem.ko.zst", |
66 | | - "/opt/open/nvidia-uvm.ko.zst", |
67 | | - "/opt/open/nvidia.ko.zst", |
68 | | - "/opt/proprietary/nvidia-drm.ko.zst", |
69 | | - "/opt/proprietary/nvidia-modeset.ko.zst", |
70 | | - "/opt/proprietary/nvidia-peermem.ko.zst", |
71 | | - "/opt/proprietary/nvidia-uvm.ko.zst", |
72 | | - "/opt/proprietary/nvidia.ko.zst", |
73 | | - ] |
| 46 | + ### TODO test kernel modules via modinfo/modprobe |
| 47 | + files += [ |
| 48 | + "/opt/open/nvidia-drm.ko.zst", |
| 49 | + "/opt/open/nvidia-modeset.ko.zst", |
| 50 | + "/opt/open/nvidia-peermem.ko.zst", |
| 51 | + "/opt/open/nvidia-uvm.ko.zst", |
| 52 | + "/opt/open/nvidia.ko.zst", |
| 53 | + "/opt/proprietary/nvidia-drm.ko.zst", |
| 54 | + "/opt/proprietary/nvidia-modeset.ko.zst", |
| 55 | + "/opt/proprietary/nvidia-peermem.ko.zst", |
| 56 | + "/opt/proprietary/nvidia-uvm.ko.zst", |
| 57 | + "/opt/proprietary/nvidia.ko.zst", |
| 58 | + ] |
74 | 59 |
|
75 | 60 | for filename in files: |
76 | 61 | assert container_per_test.connection.file(filename).exists |
0 commit comments