From 2a7219315bc998996e7269e0b5ec09dc8d7c77d6 Mon Sep 17 00:00:00 2001 From: Tim O'Donnell <timodonnell@gmail.com> Date: Fri, 13 Mar 2020 13:44:43 -0400 Subject: [PATCH] fix --- test/test_predict_scan_command.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/test/test_predict_scan_command.py b/test/test_predict_scan_command.py index a945245f..ff69b818 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: -- GitLab