def test_crop_type3_infer_neg2(self): graph = self._create_graph_type3() crop_node = Node(graph, 'crop_node') crop_node['axis'] = None with self.assertRaisesRegex(Error, "axis attribute is missing for .*"): Crop.infer(crop_node)
def test_crop_type3_infer_neg3(self): graph = self._create_graph_type3() crop_node = Node(graph, 'crop_node') crop_node['offset'] = None with self.assertRaisesRegex(Error, "offset attribute is missing.*"): Crop.infer(crop_node)
def test_crop_type2_infer_neg1(self): graph = self._create_graph_type2() crop_node = Node(graph, 'crop_node') crop_node['dim'] = int64_array([1, 2, 3]) with self.assertRaisesRegex(Error, "Number of axis.*"): Crop.infer(crop_node)
def test_crop_type1_infer_neg2(self): graph = self._create_graph_type1() crop_node = Node(graph, 'crop_node') crop_node['crop_begin'] = int64_array([1, 2, 3]) with self.assertRaisesRegex(Error, "number of crop_begin.*"): Crop.infer(crop_node)
def test_crop_type3_infer_neg4(self): graph = self._create_graph_type3() crop_node = Node(graph, 'crop_node') crop_input2 = Node(graph, 'crop_input2') crop_input2.shape = int64_array([1, 4, 423, 563]) with self.assertRaisesRegex( Error, "The crop for dimension is out of bounds.*"): Crop.infer(crop_node)
def test_crop_type3_infer_neg1(self): graph = self._create_graph_type3() crop_node = Node(graph, 'crop_node') crop_input2 = Node(graph, 'crop_input2') crop_input2.shape = None with self.assertRaisesRegex(Error, "Not all input shapes were defined.*"): Crop.infer(crop_node)
def test_crop_type2_infer_neg2(self): graph = self._create_graph_type2() crop_node = Node(graph, 'crop_node') crop_node['dim'] = None crop_node['crop_begin'] = None with self.assertRaisesRegex( Error, "Crop node crop_node should have either.*"): Crop.infer(crop_node)
def test_crop_type3_infer(self): graph = self._create_graph_type3() crop_node = Node(graph, 'crop_node') Crop.infer(crop_node) exp_shape = int64_array([1, 3, 100, 150]) res_shape = graph.node['crop_output']['shape'] self.assertTrue( np.array_equal(exp_shape, res_shape), 'shapes do not match expected: {} and given: {}'.format( exp_shape, res_shape))