fixed : Divide by zero error came when string was empty, also a test

This commit is contained in:
Manish Kumar 2016-08-26 12:52:51 +05:30
parent 51771c01c6
commit 16401d5be8
2 changed files with 4 additions and 1 deletions

View File

@ -121,6 +121,9 @@ def escaped_str_to_bytes(data):
def is_mostly_bin(s):
# type: (bytes) -> bool
if not s or len(s) == 0:
return False
return sum(
i < 9 or 13 < i < 32 or 126 < i
for i in six.iterbytes(s[:100])

View File

@ -85,7 +85,7 @@ def test_escaped_str_to_bytes():
def test_is_mostly_bin():
assert not strutils.is_mostly_bin(b"foo\xFF")
assert strutils.is_mostly_bin(b"foo" + b"\xFF" * 10)
assert not strutils.is_mostly_bin("")
def test_is_xml():
assert not strutils.is_xml(b"foo")