From 590c6038fcf5e378c562b57db12891dc2db79ea6 Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Mon, 15 May 2017 21:30:27 +0800 Subject: [PATCH] "format Test" --- python/paddle/v2/dataset/tests/mq2007_test.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/python/paddle/v2/dataset/tests/mq2007_test.py b/python/paddle/v2/dataset/tests/mq2007_test.py index c9bddddeb..59847b6c1 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() -- GitLab