#428 - Add more tests

Author: chinyeungli

tests/test_transform.py

@@ -35,7 +35,7 @@
 
 
 class TransformTest(unittest.TestCase):
-    def test_transform_data1(self):
+    def test_transform_data(self):
         test_file = get_test_loc('test_transform/input.csv')
         configuration = get_test_loc('test_transform/configuration')
         rows = read_csv_rows(test_file)
@@ -56,3 +56,15 @@ def test_transform_data_new_col(self):
         expected_data = [OrderedDict([(u'path', u'/tmp/test.c'), (u'about_resource', u'/tmp/test.c'), (u'name', u'test.c')])]
         assert col_name == expect_col
         assert data == expected_data
+
+    def test_transform_data_multi_row(self):
+        test_file = get_test_loc('test_transform/input2.csv')
+        configuration = get_test_loc('test_transform/configuration')
+        rows = read_csv_rows(test_file)
+        transformer = Transformer.from_file(configuration)
+        col_name, data, err = transform_data(rows, transformer)
+        expect_col = [u'about_resource', u'name', u'version']
+        expected_data = [OrderedDict([(u'about_resource', u'/tmp/test.c'), (u'name', u'test.c'),(u'version', u'v0.01')]),
+                         OrderedDict([(u'about_resource', u'/tmp/tmp.h'), (u'name', u'tmp.h'),(u'version', None)])]
+        assert col_name == expect_col
+        assert data == expected_data

tests/testdata/test_transform/configuration

@@ -1,3 +1,4 @@
 column_renamings:
     about_resource: 'Directory/Filename'
-    name: Component
+    name: Component
+    version: 'Confirmed Version'

tests/testdata/test_transform/input2.csv

@@ -0,0 +1,3 @@
+Directory/Filename,Component,Confirmed Version
+/tmp/test.c,test.c,v0.01
+/tmp/tmp.h,tmp.h