diff --git a/spras/btb.py b/spras/btb.py index 807dea90..62a4b161 100644 --- a/spras/btb.py +++ b/spras/btb.py @@ -161,5 +161,5 @@ def parse_output(raw_pathway_file, standardized_pathway_file): if not df.empty: df = add_rank_column(df) df = reinsert_direction_col_undirected(df) - df.columns = ['Node1', 'Node2', 'Rank', "Direction"] + df.columns = ['Node1', 'Node2', 'Rank', 'Direction'] df.to_csv(standardized_pathway_file, index=False, sep='\t', header=True) diff --git a/test/parse-outputs/expected/bowtiebuilder-pathway-expected.txt b/test/parse-outputs/expected/bowtiebuilder-pathway-expected.txt index e7b04009..b25d172b 100644 --- a/test/parse-outputs/expected/bowtiebuilder-pathway-expected.txt +++ b/test/parse-outputs/expected/bowtiebuilder-pathway-expected.txt @@ -1,8 +1,3 @@ -Node1 Node2 Rank Direction -A D 1 U -A B 1 U -C F 1 U -B C 1 U -F G 1 U -G I 1 U -G H 1 U +Node1 Node2 Rank Direction +B A 1 U +D B 1 U diff --git a/test/parse-outputs/input/bowtiebuilder-raw-pathway.txt b/test/parse-outputs/input/bowtiebuilder-raw-pathway.txt index 5699a112..8c09120f 100644 --- a/test/parse-outputs/input/bowtiebuilder-raw-pathway.txt +++ b/test/parse-outputs/input/bowtiebuilder-raw-pathway.txt @@ -1,8 +1,2 @@ -Node1 Node2 -A D -A B -C F -B C -F G -G I -G H +B A +D B