def test_squeeze(self):
x = np.linspace(0, 4 * np.pi, 101) ** (1.3)
h = np.sin(x)
surface = NonuniformLineScan(x, h).scale(2.0)
surface2 = surface.squeeze()
self.assertTrue(isinstance(surface2, NonuniformLineScan))
np.testing.assert_allclose(surface.positions(), surface2.positions())
np.testing.assert_allclose(surface.heights(), surface2.heights())
def test_positions_and_heights(self):
x = np.array((0, 1, 1.5, 2, 3))
h = 2 * x
t = NonuniformLineScan(x, h)
assert_array_equal(t.heights(), h)
assert_array_equal(t.positions(), x)
x2, h2 = t.positions_and_heights()
assert_array_equal(x2, x)
assert_array_equal(h2, h)
def test_nonuniform2(self):
x = np.array((1, 2, 3))
y = 2 * x
surf = NonuniformLineScan(x, y)
self.assertFalse(surf.is_uniform)
self.assertEqual(surf.dim, 1)
der = surf.derivative(n=1)
assert_array_equal(der, [2, 2])
der = surf.derivative(n=2)
assert_array_equal(der, [0])
surf = surf.detrend(detrend_mode='height')
self.assertFalse(surf.is_uniform)
self.assertEqual(surf.dim, 1)
der = surf.derivative(n=1)
assert_array_equal(der, [0, 0])
assert_array_equal(surf.heights(), np.zeros(y.shape))
p = surf.positions_and_heights()
assert_array_equal(p[0], x)
assert_array_equal(p[1], np.zeros(y.shape))