def test_concatenate_lines(): edf_test = EDF() test_content = ['#abc\\\n', 'abcd\\ #Comment\n', 'b\\\n', 'aa\n', 'bcd\\\n', 'aa\\\\\n', 'c\n'] output = edf_test._concatenate_lines(test_content) assert output == ['#abc\\', 'abcd b aa', 'bcd aa\\\\', 'c']
def test_read_variables(): edf_test = EDF() test_line = "Experiment: MERTIS 'MERTIS'".split() edf_test._read_variables(test_line) assert edf_test.experiment == "MERTIS 'MERTIS'" test_line = "Include: mertis.edf".split() edf_test._read_variables(test_line) output = False for files in edf_test.include_files: if 'mertis.edf' in files: output = True assert output is True test_line = "Include_file: bela.edf".split() edf_test._read_variables(test_line) output = False output2 = False for files in edf_test.include_files: if 'bela.edf' in files: output = True if 'mertis.edf' in files: output2 = True assert output is True and output2 is True test_line = "Random_variable: test".split() edf_test._read_variables(test_line) output = False for variable in edf_test.variables: if "Random_variable" in variable: if edf_test.variables["Random_variable"] == 'test': output = True assert output is True
def test_add_none_to_empty_fields(): edf_test = EDF() test_dict = {'a': ['a'], 'b': [1, 3], 'c': [1]} output = edf_test._add_none_to_empty_fields(test_dict) assert output == {'a': ['a', None], 'b': [1, 3], 'c': [1, None]} test_dict = {'a': [], 'b': [], 'c': []} output = edf_test._add_none_to_empty_fields(test_dict) assert output == {'a': [], 'b': [], 'c': []} test_dict = {'a': [1, 2], 'b': [1, 3], 'c': [1, 'a']} output = edf_test._add_none_to_empty_fields(test_dict) assert output == {'a': [1, 2], 'b': [1, 3], 'c': [1, 'a']}
def test_how_many_brackets_following(): edf_test = EDF() test_line = ['aa', 'bbadsf', '[asdf]', '[sdfs2]', 'asdfas', '[asfsddf]'] output = edf_test._how_many_brackets_following(test_line) assert output == 0 test_line = ['aa', 'bbadsf', '[asdf]', '[sdfs2]', 'asdfas', '[asfsddf]'] output = edf_test._how_many_brackets_following(test_line[2:]) assert output == 2 test_line = ['aa', 'bbadsf', '[asdf]', '[sdfs2]', 'asdfas', '[asfsddf]'] output = edf_test._how_many_brackets_following(test_line[5:]) assert output == 1