add unittest to setup.py

paddle_fl/tests/test_fl_agent.py

+from __future__ import print_function
+import paddle_fl as pfl
+import unittest
+
+class TestFLAgent(unittest.TestCase):
+    def test_recv_and_parse_kv(self):
+        pass
+
+    def test_connect_scheduler(self):
+        pass
+
+    def test_can_join_training(self):
+        pass
+
+    def test_set_sample_worker_num(self):
+        pass
+
+    def test_init_env(self):
+        pass
+
+    def test_start_fl_step(self):
+        pass
+
+if __name__ == "__main__":
+    unittest.main()

setup.py

@@ -37,6 +37,8 @@ if max_version < 3:
 else:
     REQUIRED_PACKAGES += ["numpy"]
 
+REQUIRED_PACKAGES += ["unittest"]
+
 setup(
     name='paddle_fl',
     version=fl_version.replace('-', ''),