diff --git a/python/paddle/v2/dataset/tests/mq2007_test.py b/python/paddle/v2/dataset/tests/mq2007_test.py index c9bddddeb0799c0b7f38f8afb6297f9eb1057d32..59847b6c18eadb12123cae824e8bce1051a69d4c 100644 --- a/python/paddle/v2/dataset/tests/mq2007_test.py +++ b/python/paddle/v2/dataset/tests/mq2007_test.py @@ -17,15 +17,17 @@ import unittest class TestMQ2007(unittest.TestCase): - def test_pairwise(self): - for label, query_left, query_right in paddle.v2.dataset.mq2007.test(format="pairwise"): - self.assertEqual(query_left.shape(), (46, )) - self.assertEqual(query_right.shape(), (46, )) + def test_pairwise(self): + for label, query_left, query_right in paddle.v2.dataset.mq2007.test( + format="pairwise"): + self.assertEqual(query_left.shape(), (46, )) + self.assertEqual(query_right.shape(), (46, )) - def test_listwise(self): - for label_array, query_array in paddle.v2.dataset.mq2007.test(format="listwise"): - self.assertEqual(len(label_array), len(query_array)) + def test_listwise(self): + for label_array, query_array in paddle.v2.dataset.mq2007.test( + format="listwise"): + self.assertEqual(len(label_array), len(query_array)) if __name__ == "__main__": - unittest.main() + unittest.main()