diff --git a/test/test_predict_scan_command.py b/test/test_predict_scan_command.py index a945245fa631c7d49e4b0a4ff95335d9197fcdc5..ff69b8185909b73e7efbe26296bddc905e266e71 100644 --- a/test/test_predict_scan_command.py +++ b/test/test_predict_scan_command.py @@ -17,6 +17,11 @@ setup = startup from . import data_path +def read_output_csv(filename): + return pandas.read_csv( + filename, + converters={"n_flank": str, "c_flank": str}) + def test_fasta(): args = [ @@ -31,7 +36,7 @@ def test_fasta(): full_args = args + ["--out", fd_out.name] print("Running with args: %s" % full_args) predict_scan_command.run(full_args) - result = pandas.read_csv(fd_out.name) + result = read_output_csv(fd_out.name) print(result) assert not result.isnull().any().any() finally: @@ -58,7 +63,7 @@ def test_fasta_50nm(): full_args = args + ["--out", fd_out.name] print("Running with args: %s" % full_args) predict_scan_command.run(full_args) - result = pandas.read_csv(fd_out.name) + result = read_output_csv(fd_out.name) print(result) assert not result.isnull().any().any() finally: @@ -83,7 +88,7 @@ def test_fasta_best(): full_args = args + ["--out", fd_out.name] print("Running with args: %s" % full_args) predict_scan_command.run(full_args) - result = pandas.read_csv(fd_out.name) + result = read_output_csv(fd_out.name) print(result) assert not result.isnull().any().any() finally: @@ -110,7 +115,7 @@ def test_commandline_sequences(): full_args = args + ["--out", fd_out.name] print("Running with args: %s" % full_args) predict_scan_command.run(full_args) - result = pandas.read_csv(fd_out.name) + result = read_output_csv(fd_out.name) print(result) finally: for delete in deletes: