tests

src/beignet/datasets/_random_rotation_matrix_dataset.py

@@ -1,13 +1,13 @@
 from typing import Callable, Generator
 
 import torch
-from torch.utils.data import Dataset
 
 import beignet
+from beignet.datasets.__random_rotation_dataset import RandomRotationDataset
 from beignet.transforms import Transform
 
 
-class RandomRotationMatrixDataset(Dataset):
+class RandomRotationMatrixDataset(RandomRotationDataset):
     def __init__(
         self,
         size: int,

src/beignet/datasets/_random_rotation_vector_dataset.py

@@ -1,13 +1,13 @@
 from typing import Callable, Generator
 
 import torch
-from torch.utils.data import Dataset
 
 import beignet
+from beignet.datasets.__random_rotation_dataset import RandomRotationDataset
 from beignet.transforms import Transform
 
 
-class RandomRotationVectorDataset(Dataset):
+class RandomRotationVectorDataset(RandomRotationDataset):
     def __init__(
         self,
         size: int,

tests/beignet/datasets/test__random_euler_angle_dataset.py

@@ -1,3 +1,57 @@
+import beignet
+import hypothesis.strategies
+from beignet.datasets import RandomEulerAngleDataset
+
+
+@hypothesis.strategies.composite
+def _strategy(function):
+    size = function(
+        hypothesis.strategies.integers(
+            min_value=1,
+            max_value=8,
+        ),
+    )
+
+    axes = function(
+        hypothesis.strategies.sampled_from(
+            [
+                "xyz",
+                "xzy",
+                "yxz",
+                "yzx",
+                "zxy",
+                "zyx",
+                "XYZ",
+                "XZY",
+                "YXZ",
+                "YZX",
+                "ZXY",
+                "ZYX",
+            ]
+        ),
+    )
+
+    degrees = function(hypothesis.strategies.booleans())
+
+    return (
+        {
+            "size": size,
+            "axes": axes,
+            "degrees": degrees,
+        },
+        beignet.random_euler_angle(size, axes=axes, degrees=degrees),
+    )
+
+
 class TestRandomEulerAngleDataset:
-    def test___init__(self):
-        assert True
+    @hypothesis.given(_strategy())
+    def test___init__(self, data):
+        parameters, output = data
+
+        dataset = RandomEulerAngleDataset(**parameters)
+
+        assert dataset.data.shape == output.shape
+
+        assert dataset.data.dtype == output.dtype
+
+        assert dataset.data.layout == output.layout

tests/beignet/datasets/test__random_quaternion_dataset.py

@@ -1,3 +1,37 @@
+import beignet
+import hypothesis.strategies
+from beignet.datasets import RandomQuaternionDataset
+
+
+@hypothesis.strategies.composite
+def _strategy(function):
+    size = function(
+        hypothesis.strategies.integers(
+            min_value=1,
+            max_value=8,
+        ),
+    )
+
+    canonical = function(hypothesis.strategies.booleans())
+
+    return (
+        {
+            "size": size,
+            "canonical": canonical,
+        },
+        beignet.random_quaternion(size, canonical=canonical),
+    )
+
+
 class TestRandomQuaternionDataset:
-    def test___init__(self):
-        assert True
+    @hypothesis.given(_strategy())
+    def test___init__(self, data):
+        parameters, output = data
+
+        dataset = RandomQuaternionDataset(**parameters)
+
+        assert dataset.data.shape == output.shape
+
+        assert dataset.data.dtype == output.dtype
+
+        assert dataset.data.layout == output.layout

tests/beignet/datasets/test__random_rotation_matrix_dataset.py

@@ -1,3 +1,34 @@
+import beignet
+import hypothesis.strategies
+from beignet.datasets import RandomRotationMatrixDataset
+
+
+@hypothesis.strategies.composite
+def _strategy(function):
+    size = function(
+        hypothesis.strategies.integers(
+            min_value=1,
+            max_value=8,
+        ),
+    )
+
+    return (
+        {
+            "size": size,
+        },
+        beignet.random_rotation_matrix(size),
+    )
+
+
 class TestRandomRotationMatrixDataset:
-    def test___init__(self):
-        assert True
+    @hypothesis.given(_strategy())
+    def test___init__(self, data):
+        parameters, output = data
+
+        dataset = RandomRotationMatrixDataset(**parameters)
+
+        assert dataset.data.shape == output.shape
+
+        assert dataset.data.dtype == output.dtype
+
+        assert dataset.data.layout == output.layout

tests/beignet/datasets/test__random_rotation_vector_dataset.py

@@ -1,3 +1,37 @@
+import beignet
+import hypothesis.strategies
+from beignet.datasets import RandomRotationVectorDataset
+
+
+@hypothesis.strategies.composite
+def _strategy(function):
+    size = function(
+        hypothesis.strategies.integers(
+            min_value=1,
+            max_value=8,
+        ),
+    )
+
+    degrees = function(hypothesis.strategies.booleans())
+
+    return (
+        {
+            "size": size,
+            "degrees": degrees,
+        },
+        beignet.random_rotation_vector(size, degrees=degrees),
+    )
+
+
 class TestRandomRotationVectorDataset:
-    def test___init__(self):
-        assert True
+    @hypothesis.given(_strategy())
+    def test___init__(self, data):
+        parameters, output = data
+
+        dataset = RandomRotationVectorDataset(**parameters)
+
+        assert dataset.data.shape == output.shape
+
+        assert dataset.data.dtype == output.dtype
+
+        assert dataset.data.layout == output.layout