diff --git a/python/paddle/v2/fluid/tests/test_layers.py b/python/paddle/v2/fluid/tests/test_layers.py index d40510b6b3df99164b0dd3d2c04e01a2be0d7d94..4fceb0eca5bb16cb39f9703b3a992d1936b1b2dc 100644 --- a/python/paddle/v2/fluid/tests/test_layers.py +++ b/python/paddle/v2/fluid/tests/test_layers.py @@ -170,12 +170,6 @@ class TestBook(unittest.TestCase): self.assertIsNotNone(layers.sequence_expand(x=x, y=y)) print(str(program)) - def test_get_places(self): - program = Program() - with program_guard(program): - x = layers.get_places(trainer_count=4) - print(str(program)) - def test_lstm_unit(self): program = Program() with program_guard(program): @@ -193,6 +187,12 @@ class TestBook(unittest.TestCase): x_t=x_t, hidden_t_prev=prev_hidden, cell_t_prev=prev_cell)) print(str(program)) + def test_get_places(self): + program = Program() + with program_guard(program): + x = layers.get_places(trainer_count=4) + print(str(program)) + if __name__ == '__main__': unittest.main()