From 2ce38844a27de4ab202b4af764a54a0c0477909f Mon Sep 17 00:00:00 2001 From: Kacppian Date: Sat, 28 Jul 2018 13:37:01 +0530 Subject: [PATCH] testing things out --- graphene/pyutils/version.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/graphene/pyutils/version.py b/graphene/pyutils/version.py index ab177e12..de329c43 100644 --- a/graphene/pyutils/version.py +++ b/graphene/pyutils/version.py @@ -3,6 +3,7 @@ from __future__ import unicode_literals import datetime import os import subprocess +from ..utils.comparison_helper import raise_assertion_if def get_version(version=None): @@ -44,10 +45,14 @@ def get_complete_version(version=None): if version is None: from graphene import VERSION as version else: - if len(version) is not 5: - raise AssertionError("Version needs to be 5") - if version[3] not in ("alpha", "beta", "rc", "final"): - raise AssertionError("Release version is unkown") + raise_assertion_if( + condition=len(version) is not 5, message="Version needs to be 5" + ) + + raise_assertion_if( + condition=version[3] not in ("alpha", "beta", "rc", "final"), + message="Release version is unkown", + ) return version