diff --git a/test/test_api/test_api.py b/test/test_api/test_api.py index 51c0da3b..d2ebe62b 100644 --- a/test/test_api/test_api.py +++ b/test/test_api/test_api.py @@ -169,7 +169,8 @@ def test_reference_description(Script): def test_get_line_code(Script): def get_line_code(source, line=None, **kwargs): - return Script(source).complete(line=line)[0].get_line_code(**kwargs) + # On Windows replace \r + return Script(source).complete(line=line)[0].get_line_code(**kwargs).replace('\r', '') # On builtin assert get_line_code('abs') == 'def abs(__n: SupportsAbs[_T]) -> _T: ...\n' diff --git a/test/test_inference/test_gradual/test_conversion.py b/test/test_inference/test_gradual/test_conversion.py index c77a06b0..ea9ea013 100644 --- a/test/test_inference/test_gradual/test_conversion.py +++ b/test/test_inference/test_gradual/test_conversion.py @@ -69,11 +69,12 @@ def test_stub_get_line_code(Script): code = 'from abc import ABC; ABC' script = Script(code) d, = script.goto(only_stubs=True) - assert d.get_line_code() == 'class ABC(metaclass=ABCMeta): ...\n' + # Replace \r for tests on Windows + assert d.get_line_code().replace('\r', '') == 'class ABC(metaclass=ABCMeta): ...\n' del parser_cache[script._inference_state.latest_grammar._hashed][d.module_path] d, = Script(path=d.module_path).goto(d.line, d.column, only_stubs=True) assert d.is_stub() - assert d.get_line_code() == 'class ABC(metaclass=ABCMeta): ...\n' + assert d.get_line_code().replace('\r', '') == 'class ABC(metaclass=ABCMeta): ...\n' def test_os_stat_result(Script):