1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import numpy as np
from hypothesis import given, settings
import hypothesis.strategies as st
import unittest
from caffe2.python import core
import caffe2.python.hypothesis_test_util as hu
import caffe2.python.serialized_test.serialized_test_util as serial
class TestNumpyTile(serial.SerializedTestCase):
@given(ndim=st.integers(min_value=1, max_value=4),
seed=st.integers(min_value=0, max_value=65536),
**hu.gcs_cpu_only)
@settings(deadline=10000)
def test_numpy_tile(self, ndim, seed, gc, dc):
np.random.seed(seed)
input_dims = np.random.randint(1, 4, size=ndim)
input = np.random.randn(*input_dims)
repeats = np.random.randint(1, 5, size=ndim)
op = core.CreateOperator(
'NumpyTile', ['input', 'repeats'], 'out',
)
def tile_ref(input, repeats):
tiled_data = np.tile(input, repeats)
return (tiled_data,)
# Check against numpy reference
self.assertReferenceChecks(gc, op, [input, repeats],
tile_ref)
@given(ndim=st.integers(min_value=1, max_value=4),
seed=st.integers(min_value=0, max_value=65536),
**hu.gcs_cpu_only)
@settings(deadline=10000)
def test_numpy_tile_zero_dim(self, ndim, seed, gc, dc):
np.random.seed(seed)
input_dims = np.random.randint(0, 4, size=ndim)
input = np.random.randn(*input_dims)
repeats = np.random.randint(0, 5, size=ndim)
op = core.CreateOperator(
'NumpyTile', ['input', 'repeats'], 'out',
)
def tile_ref(input, repeats):
tiled_data = np.tile(input, repeats)
return (tiled_data,)
# Check against numpy reference
self.assertReferenceChecks(gc, op, [input, repeats],
tile_ref)
if __name__ == "__main__":
unittest.main()
|
