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