test case for LinearLocator

yt/visualization/tests/test_tick_locators.py

@@ -0,0 +1,19 @@
+import numpy as np
+
+from yt.testing import assert_equal
+from yt.visualization.tick_locators import LinearLocator
+
+
+def test_linear_locator():
+    preset = {(-1.5e-10, 2.3e-10): np.linspace(-1.5e-10, 2.3e-10, 7)}
+    loc = LinearLocator(numticks=10, presets=preset)
+    desired = np.linspace(-1.2, 2.5, 10)
+    assert_equal(loc(-1.2, 2.5), desired)
+    assert_equal(loc(2.5, -1.2), desired)
+
+    desired = np.linspace(-1.5e-10, 2.3e-10, 7)
+    assert_equal(loc(-1.5e-10, 2.3e-10), desired)
+    assert_equal(loc(2.3e-10, -1.5e-10), desired)
+
+def test_log_locator():
+    pass