"format Test"

python/paddle/v2/dataset/tests/mq2007_test.py

@@ -17,15 +17,17 @@ import unittest
 
 
 class TestMQ2007(unittest.TestCase):
-  def test_pairwise(self):
+    def test_pairwise(self):
-    for label, query_left, query_right in paddle.v2.dataset.mq2007.test(format="pairwise"):
+        for label, query_left, query_right in paddle.v2.dataset.mq2007.test(
-      self.assertEqual(query_left.shape(), (46, ))
+                format="pairwise"):
-      self.assertEqual(query_right.shape(), (46, ))
+            self.assertEqual(query_left.shape(), (46, ))
+            self.assertEqual(query_right.shape(), (46, ))
 
-  def test_listwise(self):
+    def test_listwise(self):
-    for label_array, query_array in paddle.v2.dataset.mq2007.test(format="listwise"):
+        for label_array, query_array in paddle.v2.dataset.mq2007.test(
-      self.assertEqual(len(label_array), len(query_array))
+                format="listwise"):
+            self.assertEqual(len(label_array), len(query_array))
 
 
 if __name__ == "__main__":
-  unittest.main()
+    unittest.main()