From 997fcde8ce94be9d8decddd4bc783106dbb41ab3 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sat, 12 Sep 2015 17:03:09 +0200 Subject: make clean_bin unicode-aware --- test/test_utils.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) (limited to 'test') diff --git a/test/test_utils.py b/test/test_utils.py index 374d09ba..9dba5d35 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -15,10 +15,17 @@ def test_hexdump(): def test_cleanBin(): - assert utils.cleanBin("one") == "one" - assert utils.cleanBin("\00ne") == ".ne" - assert utils.cleanBin("\nne") == "\nne" - assert utils.cleanBin("\nne", True) == ".ne" + assert utils.clean_bin(b"one") == b"one" + assert utils.clean_bin(b"\00ne") == b".ne" + assert utils.clean_bin(b"\nne") == b"\nne" + assert utils.clean_bin(b"\nne", False) == b".ne" + assert utils.clean_bin(u"\u2605".encode("utf8")) == b"..." + + assert utils.clean_bin(u"one") == u"one" + assert utils.clean_bin(u"\00ne") == u".ne" + assert utils.clean_bin(u"\nne") == u"\nne" + assert utils.clean_bin(u"\nne", False) == u".ne" + assert utils.clean_bin(u"\u2605") == u"\u2605" def test_pretty_size(): -- cgit v1.2.3