BUG: set_precision format fixed (pandas-dev#13257)

BUG: set_precision format fixed (pandas-dev#13257)

Author: AntonioAndraues

pandas/io/formats/style.py

@@ -151,7 +151,8 @@ def __init__(
 
         def default_display_func(x):
             if is_float(x):
-                return "{:>.{precision}g}".format(x, precision=self.precision)
+                display_format="{0:.{precision}f}".format(x, precision=self.precision)
+                return display_format.rstrip('0').rstrip('.')
             else:
                 return x
 

pandas/tests/io/formats/test_style.py

@@ -1062,6 +1062,7 @@ def test_precision(self):
         s2 = s.set_precision(4)
         assert s is s2
         assert s.precision == 4
+    
 
     def test_apply_none(self):
         def f(x):
@@ -1156,6 +1157,33 @@ def test_display_format_raises(self):
             df.style.format(5)
         with pytest.raises(TypeError):
             df.style.format(True)
+            
+    def test_display_set_precision(self):
+        df = pd.DataFrame(data=[[1.0,2.0090],[3.2121,4.566]], columns=['a','b'])
+        s = Styler(df)
+
+        ctx = s.set_precision(1)._translate()
+
+        assert s.precision == 1
+        assert ctx["body"][0][1]["display_value"] == "1"
+        assert ctx["body"][0][2]["display_value"] == "2"
+        assert ctx["body"][1][1]["display_value"] == "3.2"
+        assert ctx["body"][1][2]["display_value"] == "4.6"
+        
+        ctx = s.set_precision(2)._translate()
+        assert s.precision == 2
+        assert ctx["body"][0][1]["display_value"] == "1"
+        assert ctx["body"][0][2]["display_value"] == "2.01"
+        assert ctx["body"][1][1]["display_value"] == "3.21"
+        assert ctx["body"][1][2]["display_value"] == "4.57"
+
+        ctx = s.set_precision(3)._translate()
+        assert s.precision == 3
+        assert ctx["body"][0][1]["display_value"] == "1"
+        assert ctx["body"][0][2]["display_value"] == "2.009"
+        assert ctx["body"][1][1]["display_value"] == "3.212"
+        assert ctx["body"][1][2]["display_value"] == "4.566"
+        # assert
 
     def test_display_subset(self):
         df = pd.DataFrame([[0.1234, 0.1234], [1.1234, 1.1234]], columns=["a", "b"])