Skip to content

Commit b96a526

Browse files
committed
Update test code
1 parent be00d90 commit b96a526

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

tests/test_transform.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,12 @@ def test_transform_data_new_col(self):
5252

5353
field_name, data, err = transform_data(data, transformer)
5454

55-
expect_col = [u'path',u'about_resource', u'name',u'version',u'notes',u'temp']
55+
expect_name = [u'path',u'about_resource', u'name',u'version',u'notes',u'temp']
5656
expected_data = [OrderedDict([(u'path', u'/tmp/test.c'), (u'about_resource', u'/tmp/test.c'), (u'name', u'test.c'),
5757
(u'version', u'1'),(u'notes', u'test'),(u'temp', u'foo')])]
58-
assert field_name == expect_col
58+
assert len(field_name) == len(expect_name)
59+
for name in field_name:
60+
assert name in expect_name
5961
assert data == expected_data
6062

6163
def test_transform_data(self):
@@ -65,12 +67,14 @@ def test_transform_data(self):
6567
configuration = get_test_loc('test_transform/configuration')
6668
transformer = Transformer.from_file(configuration)
6769

68-
col_name, data, err = transform_data(data, transformer)
70+
field_name, data, err = transform_data(data, transformer)
6971

70-
expect_col = [u'about_resource', u'name', u'version']
72+
expect_name = [u'about_resource', u'name', u'version']
7173
expected_data = [OrderedDict([(u'about_resource', u'/tmp/test.c'), (u'name', u'test.c'), (u'version', u'1')])]
7274

73-
assert col_name == expect_col
75+
assert len(field_name) == len(expect_name)
76+
for name in field_name:
77+
assert name in expect_name
7478
assert data == expected_data
7579

7680
def test_transform_data_mutli_rows(self):
@@ -79,13 +83,15 @@ def test_transform_data_mutli_rows(self):
7983
configuration = get_test_loc('test_transform/configuration2')
8084
transformer = Transformer.from_file(configuration)
8185

82-
col_name, data, err = transform_data(data, transformer)
86+
field_name, data, err = transform_data(data, transformer)
8387

84-
expect_col = [u'about_resource', u'name', u'version']
88+
expect_name = [u'about_resource', u'name', u'version']
8589
expected_data = [OrderedDict([(u'about_resource', u'/tmp/test.c'), (u'name', u'test.c'), (u'version', u'v0.01')]),
8690
OrderedDict([(u'about_resource', u'/tmp/tmp.h'), (u'name', u'tmp.h'), (u'version', None)])]
8791

88-
assert col_name == expect_col
92+
assert len(field_name) == len(expect_name)
93+
for name in field_name:
94+
assert name in expect_name
8995
assert data == expected_data
9096

9197
def test_normalize_dict_data_scancode(self):

0 commit comments

Comments
 (0)