

def test_ocl_interpolate_flow_form(self):
    fr = self.ocl_inter_method(1)[0]
    self.assertIsInstance(fr, np.ndarray)
    self.assertEqual(fr.dtype, np.uint8)
    self.assertEqual(len(fr.shape), 3)
    (r, c, ch) = fr.shape
    self.assertEqual(r, 240)
    self.assertEqual(c, 320)
    self.assertEqual(ch, 3)

dataset/ETHPy150Open dthpham/butterflow/tests/test_motion.py InterpolateFlowTestCase.test_ocl_interpolate_flow_form/original