Skip to content

Commit 15d9ac9

Browse files
committed
add zeropad2d v0.5 codestyle
1 parent 14b84b3 commit 15d9ac9

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

python/paddle/fluid/tests/unittests/test_zeropad2d.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,16 +114,21 @@ class TestZeroPad2DLayer(unittest.TestCase):
114114
"""
115115
test nn.ZeroPad2D
116116
"""
117+
117118
def setUp(self):
118119
self.shape = [4, 3, 224, 224]
119120
self.pad = [2, 2, 4, 1]
120121
self.padLayer = ZeroPad2D(padding=self.pad)
121122
self.x = np.random.randint(-255, 255, size=self.shape)
122-
self.expect_res = np.pad(
123-
self.x, [[0, 0], [0, 0], [self.pad[2], self.pad[3]], [self.pad[0], self.pad[1]]])
123+
self.expect_res = np.pad(self.x,
124+
[[0, 0], [0, 0], [self.pad[2], self.pad[3]],
125+
[self.pad[0], self.pad[1]]])
124126

125127
def test_layer(self):
126-
self.assertTrue(np.allclose(zeropad2d(to_tensor(self.x), self.pad).numpy(), self.padLayer(to_tensor(self.x))))
128+
self.assertTrue(
129+
np.allclose(
130+
zeropad2d(to_tensor(self.x), self.pad).numpy(),
131+
self.padLayer(to_tensor(self.x))))
127132

128133

129134
if __name__ == '__main__':

0 commit comments

Comments
 (0)