Add basic tests for scaling

Author: tparkerd

setup.cfg

@@ -19,6 +19,3 @@ exclude = docs
 
 [aliases]
 test = pytest
-
-[tool:pytest]
-collect_ignore = ['setup.py']

tests/test_rawtools.py

@@ -2,23 +2,66 @@
 
 """Tests for `rawtools` package."""
 
+import numpy as np
 import pytest
+from numpy import uint8, uint16
+from rawtools import rawtools
 
+DIMS = (4, 5)
 
-from rawtools import rawtools
+
+@pytest.fixture
+def slice_uint8():
+    """Sample uint8 slice"""
+    return np.rint(np.arange(0, 20, dtype=uint8).reshape(DIMS))
+
+
+@pytest.fixture
+def slice_uint16():
+    """Sample uint16 slice"""
+    return np.rint(np.arange(0, 20, dtype=uint16).reshape(DIMS))
 
 
 @pytest.fixture
-def response():
-    """Sample pytest fixture.
+def slice_uint16_high_variance():
+    """Sample uint16 slice with variable values"""
+    return np.array([-1, 0, 100, 1000, 5000, 14830, 50321, 65535, 65536], dtype=uint16)
+
+
+def test_scale_uint8(slice_uint8):
+    """Test scaling a unsigned 8-bit integer array to own bounds."""
+    from rawtools.convert import scale
+    xs = np.arange(0, 20, dtype=uint8).reshape(DIMS)
+    lbound = np.iinfo(uint8).min
+    ubound = np.iinfo(uint8).max
+    scaled_slice = scale(xs, lbound, ubound, lbound, ubound)
+    np.testing.assert_array_equal(scaled_slice, slice_uint8)
+
+
+def test_scale_uint16_to_uint8(slice_uint16):
+    """Test scaling an unsigned 16-bit integer array to an unsigned 8-bit array's bounds."""
+    from rawtools.convert import scale
+    lbound = np.iinfo(uint16).min
+    ubound = np.iinfo(uint16).max
+    new_lbound = np.iinfo(uint8).min
+    new_ubound = np.iinfo(uint8).max
+    slice_uint8 = np.zeros(DIMS, dtype=uint8)
+    scaled_slice = np.rint(
+        scale(slice_uint16, lbound, ubound, new_lbound, new_ubound))
+
+    np.testing.assert_array_equal(scaled_slice, slice_uint8)
 
-    See more at: http://doc.pytest.org/en/latest/fixture.html
-    """
-    # import requests
-    # return requests.get('https://github.com/audreyr/cookiecutter-pypackage')
 
+def test_scale_uint16_to_uint8_large_variance(slice_uint16_high_variance):
+    """Test scaling an unsigned 16-bit integer array with high variance to an unsigned 8-bit array's bounds."""
+    from rawtools.convert import scale
+    lbound = np.iinfo(uint16).min
+    ubound = np.iinfo(uint16).max
+    new_lbound = np.iinfo(uint8).min
+    new_ubound = np.iinfo(uint8).max
+    # Mapped values should wrap as they exceed target bit depth
+    slice_uint8 = np.array([255, 0, 0, 4, 19, 58, 196, 255, 0], dtype=uint8)
+    scaled_slice = np.rint(
+        scale(slice_uint16_high_variance, lbound, ubound, new_lbound, new_ubound))
 
-def test_content(response):
-    """Sample pytest test function with the pytest fixture as an argument."""
-    # from bs4 import BeautifulSoup
-    # assert 'GitHub' in BeautifulSoup(response.content).title.string
+    np.testing.assert_array_equal(scaled_slice, slice_uint8)