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