Add sampler

tests/nerf/test_sampler.py

@@ -0,0 +1,27 @@
+"""
+Test samplers
+"""
+from random import uniform
+import pytest
+import torch
+
+from mattport.nerf import sampler
+from mattport.structures.cameras import Rays
+
+
+def test_uniform_sampler():
+    """Test uniform sampler"""
+    uniform_sampler = sampler.UniformSampler()
+
+    origins = torch.zeros((10, 3))
+    directions = torch.ones_like(origins)
+    camera_rays = Rays(origin=origins, direction=directions)
+
+
+def test_pdf_sampler():
+    """Test uniform sampler"""
+    # TODO Tancik: Add more precise tests
+
+
+if __name__ == "__main__":
+    test_uniform_sampler()