Fixed tests and formatting issues

Author: fuzeman

bencode/__init__.py

@@ -14,6 +14,7 @@
 
 from bencode.BTL import BTFailure
 from bencode.exceptions import BencodeDecodeError
+
 from bencodepy import Bencached, Bencode
 
 __all__ = (

bencodepy/common.py

@@ -1,3 +1,6 @@
+"""bencode.py - common."""
+
+
 class Bencached(object):
     __slots__ = ['bencoded']
 

bencodepy/compat.py

@@ -1,3 +1,5 @@
+"""bencode.py - compatibility helpers."""
+
 import sys
 
 PY2 = sys.version_info[0] == 2

bencodepy/decoder.py

@@ -10,11 +10,10 @@
 
 # Written by Petru Paler
 
-"""bencode.py - bencode encoder + decoder."""
+"""bencode.py - bencode decoder."""
 
-from bencodepy.compat import *
+from bencodepy.compat import to_binary
 from bencodepy.exceptions import BencodeDecodeError
-
 from collections import OrderedDict
 
 try:

bencodepy/encoder.py

@@ -10,11 +10,10 @@
 
 # Written by Petru Paler
 
-"""bencode.py - bencode encoder + decoder."""
+"""bencode.py - bencode encoder."""
 
 from bencodepy.common import Bencached
-from bencodepy.compat import *
-
+from bencodepy.compat import PY2, to_binary
 from collections import deque
 
 try:

bencodepy/exceptions.py

@@ -1,4 +1,4 @@
-"""bencode.py - encoder + decode exceptions."""
+"""bencode.py - exceptions."""
 
 
 class BencodeDecodeError(Exception):

tests/bencode/__init__.py

@@ -1 +1 @@
-"""bencode.py - tests."""
+"""bencode.py - bencode tests."""

tests/bencodepy/__init__.py

@@ -0,0 +1 @@
+"""bencode.py - bencodepy tests."""

tests/bencodepy/bencode_tests.py

@@ -4,7 +4,6 @@
 """bencode.py tests."""
 
 from bencodepy import Bencached, BencodeDecodeError, bencode, bdecode
-
 import pytest
 
 try:
@@ -76,7 +75,7 @@ def test_decode_dict():
     assert isinstance(value, dict)
 
     # Validate items
-    assert value == {'title': 'Example'}
+    assert value == {b'title': b'Example'}
 
 
 def test_encode_roundtrip():

tests/bencodepy/file_tests.py

@@ -4,7 +4,6 @@
 """bencode.py - file tests."""
 
 from bencodepy import bread, bwrite
-
 import os
 import pytest
 import sys