def testJustRightFalse(self):
a = np.arange(1).reshape(1, 1)
result = shift_trajectory(a.astype(np.float64))
self.assertFalse(result[1])
b = np.arange(49).reshape(7, 7)
result2 = shift_trajectory(b.astype(np.float64))
self.assertFalse(result[1])
def testNumericFalse(self):
charar = np.chararray((3, 3))
charar[:] = 'a'
result = shift_trajectory(charar)
self.assertFalse(result[0])
def testShiftTrajectoryTrue(self):
a = np.arange(9).reshape(3, 3)
result = shift_trajectory(a.astype(np.float64))
self.assertTrue(result[0])
self.assertTrue(result[1])
