test for key value transform

acutesoftware · May 10, 2017 · 06aa9f5 · 06aa9f5
1 parent 4be7e2e
commit 06aa9f5
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/test_transpose.py b/tests/test_transpose.py
@@ -28,11 +28,21 @@ def tearDown(self):
 
     def test_01_instantiate(self):
         print(self.obj_transpose)
-        self.assertEqual(len(self.obj_transpose.data) , 5) 
+        self.assertEqual(len(self.obj_transpose.ip_data) , 5) 
 
+    def test_02_pivot(self):
+        #print(self.obj_transpose.ip_data)
 
+        self.assertEqual(self.obj_transpose.ip_data, [['col1', 'col2', 'col3'], ['r1c1', 'r1c2', 'r1c3'], ['r2c1', 'r2c2', 'r2c3'], ['r3c1', 'r3c2', 'r3c3'], ['r4c1', 'r4c2', 'r4c3']])
+        self.obj_transpose.pivot()
+        #print(self.obj_transpose.op_data)
+        self.assertEqual(self.obj_transpose.op_data, [['col1', 'r1c1', 'r2c1', 'r3c1', 'r4c1'], ['col2', 'r1c2', 'r2c2', 'r3c2', 'r4c2'], ['col3', 'r1c3', 'r2c3', 'r3c3', 'r4c3']])
 
+
+    def test_03_key_value_pairs(self):
+        self.obj_transpose.key_value_pairs()
+        #print(self.obj_transpose.op_data)
+        self.assertEqual(self.obj_transpose.op_data, [['r1c1', 'col1', 'r1c1'], ['r1c1', 'col2', 'r1c2'], ['r1c1', 'col3', 'r1c3'], ['r2c1', 'col1', 'r2c1'], ['r2c1', 'col2', 'r2c2'], ['r2c1', 'col3', 'r2c3'], ['r3c1', 'col1', 'r3c1'], ['r3c1', 'col2', 'r3c2'], ['r3c1', 'col3', 'r3c3'], ['r4c1', 'col1', 'r4c1'], ['r4c1', 'col2', 'r4c2'], ['r4c1', 'col3', 'r4c3']])
 
-
 if __name__ == '__main__':
     unittest.main()