def testJustRightFalse(self):
a = np.arange(9).reshape(3, 3)
b = np.arange(1).reshape(1, 1)
c = np.arange(49).reshape(7, 7)
result = plot_trajectory(b.astype(np.float64), 'Plot 1')
self.assertFalse(result)
result2 = plot_trajectory(c.astype(np.float64), 'Plot 1')
self.assertFalse(result2)
def testPlotTrue(self):
a = np.arange(9).reshape(3, 3)
result = plot_trajectory(a.astype(np.float64), 'Plot 1')
self.assertTrue(result)
