refine unittest, test=develop

revert-15774-anakin_subgraph_engine
jerrywgz 6 years ago
parent 8fc0fc314a
commit a52d5d5095

@ -112,7 +112,10 @@ class TestExpandOpRank4(OpTest):
class TestExpandOpInteger(OpTest): class TestExpandOpInteger(OpTest):
def setUp(self): def setUp(self):
self.op_type = "expand" self.op_type = "expand"
self.inputs = {'X': np.random.random((2, 4, 5)).astype("int32")} self.inputs = {
'X': np.random.randint(
10, size=(2, 4, 5)).astype("int32")
}
self.attrs = {'expand_times': [2, 1, 4]} self.attrs = {'expand_times': [2, 1, 4]}
output = np.tile(self.inputs['X'], (2, 1, 4)) output = np.tile(self.inputs['X'], (2, 1, 4))
self.outputs = {'Out': output} self.outputs = {'Out': output}
@ -124,7 +127,7 @@ class TestExpandOpInteger(OpTest):
class TestExpandOpBoolean(OpTest): class TestExpandOpBoolean(OpTest):
def setUp(self): def setUp(self):
self.op_type = "expand" self.op_type = "expand"
self.inputs = {'X': np.random.random((2, 4, 5)).astype("bool")} self.inputs = {'X': np.random.randint(2, size=(2, 4, 5)).astype("bool")}
self.attrs = {'expand_times': [2, 1, 4]} self.attrs = {'expand_times': [2, 1, 4]}
output = np.tile(self.inputs['X'], (2, 1, 4)) output = np.tile(self.inputs['X'], (2, 1, 4))
self.outputs = {'Out': output} self.outputs = {'Out': output}

Loading…
Cancel
Save