TST: verify handling of large-stride np.arrays

Author: pavoljuhas

src/diffpy/srreal/tests/testpdfbaseline.py

@@ -53,7 +53,10 @@ def test___call__(self):
         self.assertEqual(-7.0, self.linear(3.5))
         self.assertEqual(-2.0, self.linear._getDoubleAttr('slope'))
         x = numpy.arange(0, 10.001, 0.1)
+        xb = numpy.array([(0.0, xi) for xi in x])[:, 1]
+        self.assertTrue(xb.strides > x.strides)
         self.assertTrue(numpy.array_equal(-2 * x, self.linear(x)))
+        self.assertTrue(numpy.array_equal(-2 * x, self.linear(xb)))
         return
 
 

src/diffpy/srreal/tests/testpdfenvelope.py

@@ -43,12 +43,15 @@ def test___call__(self):
         """check PDFEnvelope.__call__()
         """
         x = numpy.arange(0, 9.1, 0.3)
+        xb = numpy.array([(0.0, xi) for xi in x])[:, 1]
+        self.assertTrue(xb.strides > x.strides)
         # this is a virtual method in the base class
         self.assertRaises(RuntimeError, PDFEnvelope().__call__, 37)
         self.assertEqual(0.0, self.fstepcut(10))
         self.assertEqual(1.0, self.fstepcut(3.45))
         ycheck = numpy.array(17 * [1] + 14 * [0])
         self.assertTrue(numpy.array_equal(ycheck, self.fstepcut(x)))
+        self.assertTrue(numpy.array_equal(ycheck, self.fstepcut(xb)))
         self.assertEqual(1.0, self.fscale(3.45))
         self.assertEqual(1.0, self.fscale(345))
         self.assertTrue(numpy.array_equal(numpy.ones_like(x), self.fscale(x)))

src/diffpy/srreal/tests/testscatteringfactortable.py

@@ -167,10 +167,12 @@ def test_lookup(self):
         """Check ScatteringFactorTable.lookup handling of array arguments.
         """
         qa = numpy.linspace(0, 50)
+        qb = numpy.array([(x, 0.0) for x in qa])[:,0]
+        self.assertTrue(qb.strides > qa.strides)
         sftx = self.sftx
         fmn0 = numpy.array([sftx.lookup('Mn', x) for x in qa])
-        fmn1 = sftx.lookup('Mn', qa)
-        self.assertTrue(numpy.array_equal(fmn0, fmn1))
+        self.assertTrue(numpy.array_equal(fmn0, sftx.lookup('Mn', qa)))
+        self.assertTrue(numpy.array_equal(fmn0, sftx.lookup('Mn', qb)))
         self.assertTrue(numpy.array_equal(
             fmn0.reshape(5, 10), sftx.lookup('Mn', qa.reshape(5, 10))))
         self.assertTrue(numpy.array_equal(