Skip to content

Commit

Permalink
Fix FloatRationalizer() for CPython 3.9
Browse files Browse the repository at this point in the history
  • Loading branch information
skirpichev committed Feb 4, 2021
1 parent 14ad30a commit 5599736
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
3 changes: 2 additions & 1 deletion diofant/interactive/session.py
Expand Up @@ -69,12 +69,13 @@ def visit_Name(self, node):
class FloatRationalizer(ast.NodeTransformer):
"""Wraps all floats in a call to Rational."""

def visit_Num(self, node):
def visit_Constant(self, node):
if isinstance(node.n, float):
return ast.Call(func=ast.Name(id='Rational', ctx=ast.Load()),
args=[ast.Str(s=repr(node.n))], keywords=[],
starargs=None, kwargs=None)
return node
visit_Num = visit_Constant

def visit_Call(self, node):
if isinstance(node.func, ast.Name) and node.func.id == 'Float':
Expand Down
3 changes: 0 additions & 3 deletions diofant/tests/interactive/test_ipythonprinting.py
@@ -1,7 +1,5 @@
"""Tests that the IPython printing module is properly loaded."""

import sys

import pytest

from diofant.core import Float, Rational, Symbol
Expand All @@ -12,7 +10,6 @@
ipython = pytest.importorskip('IPython', minversion='2.3.0')


@pytest.mark.skipif(sys.version_info >= (3, 8), reason='Broken on 3.8')
def test_ipython_printing(monkeypatch):
app = ipython.terminal.ipapp.TerminalIPythonApp()
app.display_banner = False
Expand Down

0 comments on commit 5599736

Please sign in to comment.