From 65c3309e94c863a7b30a5015b5eab4774178e92b Mon Sep 17 00:00:00 2001
From: Paul Makepeace <Paul.Makepeace@realprogrammers.com>
Date: Mon, 25 Apr 2011 23:15:35 -0400
Subject: [PATCH] Add reorder_columns(), transpose_rows_into_columns()

---
 google/refine.py           | 16 ++++++++-
 google/test/test_refine.py | 70 ++++++++++++++++++++++++++++++++++++--
 2 files changed, 82 insertions(+), 4 deletions(-)

diff --git a/google/refine.py b/google/refine.py
index beb9c0c..aaab918 100644
--- a/google/refine.py
+++ b/google/refine.py
@@ -569,7 +569,14 @@ class RefineProject:
 
     def rename_column(self, column, new_column):
         response = self.do_json('rename-column', {'oldColumnName': column,
-            'newColumnName': new_column})
+                                                  'newColumnName': new_column})
+        self.get_models()
+        return response
+
+    def reorder_columns(self, new_column_order):
+        """Takes an array of column names in the new order."""
+        response = self.do_json('reorder-columns', {
+            'columnNames': new_column_order})
         self.get_models()
         return response
 
@@ -593,3 +600,10 @@ class RefineProject:
             'separator': separator, 'ignoreBlankCells': ignore_blank_cells})
         self.get_models()
         return response
+
+    def transpose_rows_into_columns(self, column, row_count):
+        response = self.do_json('transpose-rows-into-columns', {
+            'columnName': column, 'rowCount': row_count})
+        self.get_models()
+        return response
+
diff --git a/google/test/test_refine.py b/google/test/test_refine.py
index 811741c..a8c0358 100644
--- a/google/test/test_refine.py
+++ b/google/test/test_refine.py
@@ -42,6 +42,7 @@ class CamelTest(unittest.TestCase):
 
 class RefineTestCase(unittest.TestCase):
     project_file = None
+    project_file_options = {}
     project = None
     # Section "2. Exploration using Facets": {1}, {2}
     def setUp(self):
@@ -49,7 +50,8 @@ class RefineTestCase(unittest.TestCase):
         self.refine = Refine(self.server)
         if self.project_file:
             self.project = self.refine.new_project(
-                os.path.join(PATH_TO_TEST_DATA, self.project_file))
+                os.path.join(PATH_TO_TEST_DATA, self.project_file),
+                **self.project_file_options)
 
     def tearDown(self):
         if self.project:
@@ -359,9 +361,71 @@ class TutorialTestTransposeColumnsIntoRows(RefineTestCase):
 
 class TutorialTestTransposeFixedNumbeOfRowsIntoColumns(RefineTestCase):
     project_file = 'fixed-rows.csv'
-
+    project_file_options = {'split_into_columns': False,
+                            'header_lines': 0}
     def test_transpose_fixed_number_of_rows_into_columns(self):
-        pass
+        # Section "5. Structural Editing,
+        #             Transpose Fixed Number of Rows into Columns"
+        # {1}
+        self.assertTrue('Column' in self.project.column_order)
+        # {8}
+        response = self.project.transpose_rows_into_columns('Column', 4)
+        self.assertTrue('Transpose every 4 cells in column Column' in
+                        response['historyEntry']['description'])
+        # {9} - renaming column triggers a bug in Refine
+        # {10}
+        response = self.project.add_column('Column 1', 'Transaction',
+            'if(value.contains(" sent "), "send", "receive")')
+        self.assertTrue('Column 1 by filling 4 rows' in
+                        response['historyEntry']['description'])
+        # {11}
+        transaction_facet = TextFacet(column='Transaction', selection='send')
+        self.project.engine.add_facet(transaction_facet)
+        self.project.compute_facets()
+        # {12}, {13}, {14}
+        response = self.project.add_column('Column 1', 'Sender',
+            'value.partition(" sent ")[0]')
+        # XXX resetting the facet shows data in rows with Transaction=receive
+        #     which shouldn't have been possible with the facet.
+        response = self.project.add_column('Column 1', 'Recipient',
+                'value.partition(" to ")[2].partition(" on ")[0]')
+        response = self.project.add_column('Column 1', 'Amount',
+                'value.partition(" sent ")[2].partition(" to ")[0]')
+        # {15}
+        transaction_facet.reset().include('receive')
+        response = self.project.get_rows()
+        # XXX there seems to be some kind of bug where the model doesn't
+        #     match get_rows() output - cellIndex being returned that are
+        #     out of range.
+        #self.assertTrue(a_row['Sender'] is None)
+        #self.assertTrue(a_row['Recipient'] is None)
+        #self.assertTrue(a_row['Amount'] is None)
+        # {16}
+        for column, expression in (
+            ('Sender',
+             'cells["Column 1"].value.partition(" from ")[2]'
+              '.partition(" on ")[0]'),
+            ('Recipient',
+             'cells["Column 1"].value.partition(" received ")[0]'),
+            ('Amount',
+             'cells["Column 1"].value.partition(" received ")[2]'
+             '.partition(" from ")[0]')
+        ):
+            response = self.project.text_transform(column, expression)
+            self.assertTrue('2 cells' in
+                            response['historyEntry']['description'])
+        # {17}
+        transaction_facet.reset()
+        # {18}
+        response = self.project.text_transform('Column 1',
+                                               'value.partition(" on ")[2]')
+        self.assertTrue('4 cells' in
+                        response['historyEntry']['description'])
+        # {19}
+        response = self.project.reorder_columns([
+            'Transaction', 'Amount', 'Sender', 'Recipient'])
+        self.assertEqual('Reorder columns',
+                         response['historyEntry']['description'])
 
 
 class TutorialTestTransposeVariableNumbeOfRowsIntoColumns(RefineTestCase):