mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 10:34:29 +08:00
17 lines
498 B
Python
17 lines
498 B
Python
"""
|
|
Test the camera classes.
|
|
"""
|
|
|
|
import torch
|
|
|
|
from pyrad.structures.cameras import PinholeCamera
|
|
from pyrad.structures.rays import RayBundle
|
|
|
|
|
|
def test_pinhole_camera():
|
|
"""Test that the pinhole camera model works."""
|
|
pinhole_camera = PinholeCamera(cx=400.0, cy=400.0, fx=10.0, fy=10.0)
|
|
camera_ray_bundle = pinhole_camera.generate_camera_rays()
|
|
assert isinstance(camera_ray_bundle, RayBundle)
|
|
assert torch.allclose(camera_ray_bundle.origins[0], torch.tensor([0.0, 0.0, 0.0]))
|