use simpleutils to save work

.travis.yml

@@ -40,6 +40,7 @@ install:
   - pip install coveralls
   - pip install cvxopt
   - pip install cvxpy
+  - pip install simpleutils
 # command to run tests
 script:
   - python setup.py build_ext --inplace

alphamind/tests/test_suite.py

@@ -5,12 +5,12 @@ Created on 2017-4-25
 @author: cheng.li
 """
 
-import os
-import sys
+from simpleutils import add_parent_path
 
-current_path = os.path.abspath(__file__)
-sys.path.append(os.path.sep.join(current_path.split(os.path.sep)[:-3]))
+add_parent_path(__file__, 3)
 
+from simpleutils import TestRunner
+from alphamind.utilities import alpha_logger
 from alphamind.tests.data.test_neutralize import TestNeutralize
 from alphamind.tests.data.test_standardize import TestStandardize
 from alphamind.tests.data.test_winsorize import TestWinsorize
@@ -18,8 +18,6 @@ from alphamind.tests.portfolio.test_rankbuild import TestRankBuild
 from alphamind.tests.portfolio.test_percentbuild import TestPercentBuild
 from alphamind.tests.portfolio.test_linearbuild import TestLinearBuild
 from alphamind.tests.settlement.test_simplesettle import TestSimpleSettle
-from alphamind.utilities import alpha_logger
-from alphamind.utilities import TestRunner
 
 
 if __name__ == '__main__':

alphamind/utilities.py

@@ -5,51 +5,13 @@ Created on 2017-4-25
 @author: cheng.li
 """
 
-import os
-import sys
-import logging
-import unittest
 import math
-from typing import List
+from simpleutils import CustomLogger
 import numpy as np
 import numba as nb
 
 
-alpha_logger = logging.getLogger('ALPHA_MIND')
-ch = logging.StreamHandler()
-formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-ch.setFormatter(formatter)
-alpha_logger.addHandler(ch)
-alpha_logger.setLevel(logging.INFO)
-
-
-def add_parent_path(name: str, level: int) -> None:
-    current_path = os.path.abspath(name)
-    sys.path.append(os.path.sep.join(current_path.split(os.path.sep)[:-level]))
-
-
-class TestRunner(object):
-
-    def __init__(self,
-                 test_cases,
-                 logger):
-
-        self.suite = unittest.TestSuite()
-        self.logger = logger
-
-        for case in test_cases:
-            tests = unittest.TestLoader().loadTestsFromTestCase(case)
-            self.suite.addTests(tests)
-
-    def run(self):
-
-        self.logger.info('Python ' + sys.version)
-
-        res = unittest.TextTestRunner(verbosity=3).run(self.suite)
-        if len(res.errors) >= 1 or len(res.failures) >= 1:
-            sys.exit(-1)
-        else:
-            sys.exit(0)
+alpha_logger = CustomLogger('ALPHA_MIND', 'info')
 
 
 @nb.njit(nogil=True, cache=True)

requirements.txt

@@ -5,4 +5,5 @@ numpy >= 1.12.1
 numba >= 0.30.0
 scikit-learn >= 0.18.1
 scipy >= 0.19.0
+simpleutils >= 0.1.0
 pandas >= 0.19.2
\ No newline at end of file

setup.py

@@ -58,7 +58,7 @@ setup(
     version=VERSION,
     packages=find_packages(),
     url='',
-    license='',
+    license='MIT',
     author='wegamekinglc',
     author_email='',
     ext_modules=ext_modules_settings,