Merge pull request #122 from free-soellingeraj/soellingeraj/version-travis-checker

Soellingeraj/version travis checker

Author: DanielGoldfarb

.travis.yml

@@ -8,4 +8,6 @@ matrix:
     - python: 3.8
 install:
   - pip install tox-travis
-script: tox
+script: 
+  - tox
+  - /bin/bash scripts/check_version.sh $TRAVIS_PULL_REQUEST

scripts/check_version.py

@@ -0,0 +1,39 @@
+import argparse
+from packaging import version
+
+
+def compare_versions(v_str1, v_str2):
+    """
+    Compares two version strings.
+    
+    In:
+        v_str1: str, __version__ string
+        v_str2: str, __version__ string
+    
+    Returns:
+        '>' | '==' | '<', reads v_str1 (x) v_str2
+    """
+    
+    p_str1 = version.parse(v_str1)
+    p_str2 = version.parse(v_str2)
+    if p_str1 > p_str2:
+        return '>'
+    
+    if p_str1 == p_str2:
+        return '=='
+    
+    return '<'
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Check versions.')
+    parser.add_argument('--pr_v', type=str, help='mplfinance.__version__ for the PR')
+    parser.add_argument('--in_v', type=str, help='mplfinance.__version__ for the `master` branch')
+    
+    args = parser.parse_args()
+    test_tag = 'VersionCheck:{}'
+    outcome = compare_versions(
+        v_str1=args.pr_v, 
+        v_str2=args.in_v
+    )
+    print(test_tag.format('pr{}master'.format(outcome)))

scripts/check_version.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+# Run this script test.sh from project root directory containing setup.py
+
+set -ev
+
+pr_branch=$1
+if [ "${pr_branch}" == "false" ]
+# All integrations pass if they're merged without a PR.
+then
+    exit 0
+fi
+
+echo "PR: branch passed in: $pr_branch"
+git fetch origin +refs/pull/${pr_branch}/merge
+git checkout FETCH_HEAD
+pip3 install .
+pr_version=$(python3 -c "import mplfinance; print(mplfinance.__version__)")
+git checkout master
+pip3 install .
+in_version=$(python3 -c "import mplfinance; print(mplfinance.__version__)")
+echo "PR: ${pr_version}; Incumbent: ${in_version}"
+
+# test the versions
+git checkout FETCH_HEAD
+pip3 install .
+result=$(python3 scripts/check_version.py --pr_v ${pr_version} --in_v ${in_version})
+if [ "${result}" != "VersionCheck:pr>master" ]
+then
+    # version in PR doesn't pass the test
+    echo "${result}"
+    exit 1
+fi
+echo "VersionCheck PASSED with response: ${result}"

src/mplfinance/_version.py

@@ -1,6 +1,7 @@
-version_info = (0, 12, 4, 'alpha', 0)
 
-_specifier_ = {'alpha': 'a', 'beta': 'b', 'candidate': 'rc', 'final': ''}
+version_info = (0, 12, 4, 'alpha', 1)
+
+_specifier_ = {'alpha': 'a','beta': 'b','candidate': 'rc','final': ''}
 
 __version__ = '%s.%s.%s%s'%(version_info[0], version_info[1], version_info[2],
-  '' if version_info[3]=='final' else _specifier_[version_info[3]]+str(version_info[4]))
+  '' if version_info[3]=='final' else _specifier_[version_info[3]]+str(version_info[4]))