def test_4(self):
graph = build_graph(nodes_attributes, edges,
{'slice_like': {
'axes': (-1, )
}})
slice_like = Node(graph, 'slice_like')
SliceLike.infer(slice_like)
ref_shape = int64_array([3, 3])
res_shape = graph.node['out_data']['shape']
self.assertTrue(np.array_equal(res_shape, ref_shape))
def test_3(self):
graph = build_graph(nodes_attributes, edges,
{'slice_like': {
'axes': (0, )
}})
slice_like = Node(graph, 'slice_like')
SliceLike.infer(slice_like)
ref_shape = int64_array([2, 4])
ref_value = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
res_shape = graph.node['out_data']['shape']
res_value = graph.node['out_data']['value']
self.assertTrue(np.array_equal(res_shape, ref_shape))
self.assertTrue(np.array_equal(res_value, ref_value))