From cc21eac88a07ba52da0c4352c0f9f16f6b1af26f Mon Sep 17 00:00:00 2001 From: Richard Hudson Date: Wed, 29 Dec 2021 13:33:56 +0100 Subject: [PATCH] Use \n rather than linesep for consistency with wasabi --- spacy/tests/test_visualization.py | 10 ++-------- spacy/visualization.py | 3 +-- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/spacy/tests/test_visualization.py b/spacy/tests/test_visualization.py index 1b91cd5e9..99b227086 100644 --- a/spacy/tests/test_visualization.py +++ b/spacy/tests/test_visualization.py @@ -646,7 +646,6 @@ def test_visualization_spacing( Visualizer() .render_table(fully_featured_doc_one_sentence, formats, spacing=1) .strip() - .replace("\r", "") == """ ╔>╔═ poss Sarah sarah PROPN NNP NounType=prop|Number=sing PERSON ║ ╚> case 's 's PART POS Poss=yes @@ -658,9 +657,7 @@ def test_visualization_spacing( ╠══>╔═ prep via via ADP IN ║ ╚> pobj London london PROPN NNP NounType=prop|Number=sing GPE ╚════> punct . . PUNCT . PunctType=peri - """.strip().replace( - "\r", "" - ) + """.strip() ) @@ -682,7 +679,6 @@ def test_visualization_minimal_render_table_two_sentences( Visualizer() .render_table(fully_featured_doc_two_sentences, formats, spacing=3) .strip() - .replace("\r", "") == """ ╔>╔═ poss Sarah sarah PROPN NNP NounType=prop|Number=sing PERSON ║ ╚> case 's 's PART POS Poss=yes @@ -700,9 +696,7 @@ def test_visualization_minimal_render_table_two_sentences( ╠═ ROOT loved love VERB VBD Tense=Past|VerbForm=Fin ╠> dobj it it PRON PRP Case=Acc|Gender=Neut|Number=Sing|Person=3|PronType=Prs ╚> punct . . PUNCT . PunctType=peri -""".strip().replace( - "\r", "" - ) +""".strip() ) diff --git a/spacy/visualization.py b/spacy/visualization.py index e41358e74..d21084e2b 100644 --- a/spacy/visualization.py +++ b/spacy/visualization.py @@ -1,5 +1,4 @@ from typing import Dict, List, Optional, Union, cast -from os import linesep import wasabi from wasabi.util import supports_ansi from spacy.tokens import Span, Token, Doc @@ -431,7 +430,7 @@ class Visualizer: bg_colors=bg_colors, spacing=spacing, ) - + linesep + + "\n" ) return return_string