fix string.rjust use for python3

1 files changed, 4 insertions, 1 deletions

diff --git a/tests/test_colors.py b/tests/test_colors.py
index 3cfedcb..edf51e9 100644
--- a/tests/test_colors.py
+++ b/tests/test_colors.py
@@ -85,6 +85,9 @@ def test_colordiff():
 
 
 def test_justify_formatted():
+    def rjust(s, width):
+        return s.rjust(width)
+
     assert justify_formatted(
-        red("hi"), string.rjust, 10
+        red("hi"), rjust, 10
     ) == "        " + red("hi")