Fix repr(float) when given integral numbers - fixes #103

Author: Tim-St

py/float.go

@@ -48,6 +48,9 @@ func FloatNew(metatype *Type, args Tuple, kwargs StringDict) (Object, error) {
 }
 
 func (a Float) M__str__() (Object, error) {
+	if i := int64(a); Float(i) == a {
+		return String(fmt.Sprintf("%d.0", i)), nil
+	}
 	return String(fmt.Sprintf("%g", a)), nil
 }
 

py/tests/float.py

@@ -14,4 +14,24 @@
 assert float(" -1E400") == float("-inf")
 assertRaises(ValueError, float, "1 E200")
 
+doc="repr"
+assert repr(float("1.0")) == "1.0"
+assert repr(float("1.")) == "1.0"
+assert repr(float("1.1")) == "1.1"
+assert repr(float("1.11")) == "1.11"
+assert repr(float("-1.0")) == "-1.0"
+assert repr(float("1.00101")) == "1.00101"
+assert repr(float("1.00")) == "1.0"
+assert repr(float("2.010")) == "2.01"
+
+doc="str"
+assert str(float("1.0")) == "1.0"
+assert str(float("1.")) == "1.0"
+assert str(float("1.1")) == "1.1"
+assert str(float("1.11")) == "1.11"
+assert str(float("-1.0")) == "-1.0"
+assert str(float("1.00101")) == "1.00101"
+assert str(float("1.00")) == "1.0"
+assert str(float("2.010")) == "2.01"
+
 doc="finished"

py/tests/list.py

@@ -9,12 +9,14 @@
 assert str([1,2,3]) == "[1, 2, 3]"
 assert str([1,[2,3],4]) == "[1, [2, 3], 4]"
 assert str(["1",[2.5,17,[]]]) == "['1', [2.5, 17, []]]"
+assert str([1, 1.0]) == "[1, 1.0]"
 
 doc="repr"
 assert repr([]) == "[]"
 assert repr([1,2,3]) == "[1, 2, 3]"
 assert repr([1,[2,3],4]) == "[1, [2, 3], 4]"
 assert repr(["1",[2.5,17,[]]]) == "['1', [2.5, 17, []]]"
+assert repr([1, 1.0]) == "[1, 1.0]"
 
 doc="enumerate"
 a = [e for e in enumerate([3,4,5,6,7], 4)]

py/tests/tuple.py

@@ -7,12 +7,14 @@
 assert str((1,2,3)) == "(1, 2, 3)"
 assert str((1,(2,3),4)) == "(1, (2, 3), 4)"
 assert str(("1",(2.5,17,()))) == "('1', (2.5, 17, ()))"
+assert str((1, 1.0)) == "(1, 1.0)"
 
 doc="repr"
 assert repr(()) == "()"
 assert repr((1,2,3)) == "(1, 2, 3)"
 assert repr((1,(2,3),4)) == "(1, (2, 3), 4)"
 assert repr(("1",(2.5,17,()))) == "('1', (2.5, 17, ()))"
+assert repr((1, 1.0)) == "(1, 1.0)"
 
 doc="mul"
 a = (1, 2, 3)