def test_csv_str_to_json_str_spacing(self): csv_str1 = 'a=1,b=2,c=3' csv_str2 = 'a = 1, b = 2, c = 3' json_str = '{a : 1, b : 2, c : 3}' converted_csv_str1 = params_dict.nested_csv_str_to_json_str(csv_str1) converted_csv_str2 = params_dict.nested_csv_str_to_json_str(csv_str2) self.assertEqual(converted_csv_str1, converted_csv_str2) self.assertEqual(converted_csv_str1, json_str) self.assertEqual(converted_csv_str2, json_str)
def test_csv_str_load_supported_datatypes(self): csv_str = 'a=1,b=2.,c=[1,2,3],d=\'hello, there\',e=\"Hi.\"' converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) converted_dict = yaml.load(converted_csv_str) self.assertEqual(converted_dict['a'], 1) self.assertEqual(converted_dict['b'], 2.) self.assertEqual(converted_dict['c'], [1, 2, 3]) self.assertEqual(converted_dict['d'], 'hello, there') self.assertEqual(converted_dict['e'], 'Hi.')
def test_gcs_added_quotes(self): csv_str = 'a=gs://abc, b=gs://def' expected_output = '{a : \'gs://abc\', b : \'gs://def\'}' converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) self.assertEqual(converted_csv_str, expected_output)
def test_complex_nested_csv_str_load(self): csv_str = 'a.aa.aaa.aaaaa.a=1,a.a=2' expected_output = {'a': {'aa': {'aaa': {'aaaaa': {'a': 1}}}, 'a': 2}} converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) converted_dict = yaml.load(converted_csv_str) self.assertDictEqual(converted_dict, expected_output)
def test_complex_nested_csv_str_to_json_str(self): csv_str = 'a.aa.aaa.aaaaa.a=1' json_str = '{a : {aa : {aaa : {aaaaa : {a : 1}}}}}' converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) self.assertEqual(converted_csv_str, json_str)
def test_basic_nested_csv_str_load(self): csv_str = 'a=1,b.b1=2,c.c1=3' expected_output = {'a': 1, 'b': {'b1': 2}, 'c': {'c1': 3}} converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) converted_dict = yaml.load(converted_csv_str) self.assertDictEqual(converted_dict, expected_output)
def test_basic_nested_csv_str_to_json_str(self): csv_str = 'a=1,b.b1=2' json_str = '{a : 1, b : {b1 : 2}}' converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) self.assertEqual(converted_csv_str, json_str)
def test_basic_csv_str_to_json_str(self): csv_str = 'a=1,b=2,c=3' json_str = '{a : 1, b : 2, c : 3}' converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str) self.assertEqual(converted_csv_str, json_str)