Add tests for bencode

Author: whtsky

.travis.yml

@@ -10,5 +10,4 @@ python:
 
 before_install: pip install Cython
 install: python setup.py install
-script: coverage run --source=bencoder setup.py -q nosetests
-after_success: coveralls
+script: nosetests

tests/test_encode.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+
+from bencoder import bencode
+
+
+def test_encode_str():
+    assert bencode("WWWWWW") == b'6:WWWWWW'
+
+
+def test_encode_int():
+    assert bencode(233) == b'i233e'
+
+
+def test_encode_bytes():
+    b = b"TheseAreSomeBytes"
+    coded = bencode(b)
+    l = bytes(str(len(b)))
+    assert coded == l + b':' + b
+
+
+def test_encode_list():
+    assert bencode(['a', 'b', 3]) == b'l1:a1:bi3ee'
+
+
+def test_encode_tuple():
+    assert bencode(('a', 'b', 3)) == b'l1:a1:bi3ee'
+
+
+def test_encode_dict():
+    od = dict()
+    od['ka'] = 'va'
+    od['kb'] = 2
+    assert bencode(od) == b'd2:ka2:va2:kbi2ee'
+
+
+def test_encode_complex():
+    od = dict()
+    od['KeyA'] = ['listitemA', {'k': 'v'}, 3]
+    od['KeyB'] = {'k': 'v'}
+    od['KeyC'] = 3
+    od['KeyD'] = 'AString'
+    expected_result = b'd4:KeyAl9:listitemAd1:k1:vei3ee4:KeyBd1:k1:ve4:KeyCi3e4:KeyD7:AStringe'
+    assert bencode(od) == expected_result