From 780f78c840fe3bb84d276cf33026fa6e092d7268 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Thu, 13 Jun 2019 16:40:23 +0200
Subject: [PATCH] Add test suite to setup.py

---
 README.md | 5 +----
 setup.py  | 1 +
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index c42703e..8b95521 100644
--- a/README.md
+++ b/README.md
@@ -153,12 +153,9 @@ print(data == cbc_decrypted)  # True
 
 ## Testing
 
-1. Install the package: `pip3 install -U tgcrypto`.
 2. Clone this repository: `git clone https://github.com/pyrogram/tgcrypto`.
 3. Enter the directory: `cd tgcrypto`.
-4. Run tests: `python3 -m unittest`.
-
-**Note**: I haven't found any public test vectors for AES256-IGE, yet.
+4. Run tests: `python3 setup.py test`.
 
 ## License
 
diff --git a/setup.py b/setup.py
index a929860..aeb54c9 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@ setup(
     },
     python_requires="~=3.4",
     packages=find_packages(),
+    test_suite="tests",
     zip_safe=False,
     ext_modules=[
         Extension(