update imports for python3 support

Author: weallwegot

gre2gmat/conversions.py

@@ -1,106 +1,107 @@
+from __future__ import absolute_import
 import warnings
-from data_table import score_table
-from constants import MAX_GRE, MIN_GRE, BAD_INPUT_RETURN, MIN_GMAT, MAX_GMAT
-
-
-def _validate_score_input(v,q):
-	try:
-		verbal = int(v)
-	except:
-		warnings.warn("Was not able to parse verbal score: {}\nReturning {}".format(v,BAD_INPUT_RETURN))
-		return BAD_INPUT_RETURN,BAD_INPUT_RETURN
-	try:
-		quant = int(q)
-	except:
-		warnings.warn("Was not able to parse quant score: {}\nReturning {}".format(q,BAD_INPUT_RETURN))
-		return BAD_INPUT_RETURN, BAD_INPUT_RETURN
-
-	if verbal > MAX_GRE or verbal < MIN_GRE:
-		warnings.warn("Verbal score of {} is not between {} and {}\nReturning {}".format(verbal,MIN_GRE,MAX_GRE,BAD_INPUT_RETURN))
-		return BAD_INPUT_RETURN, BAD_INPUT_RETURN
-
-	if quant > MAX_GRE or quant < MIN_GRE:
-		warnings.warn("Quant score of {} is not between {} and {}\nReturning {}".format(quant,MIN_GRE,MAX_GRE,BAD_INPUT_RETURN))
-		return BAD_INPUT_RETURN, BAD_INPUT_RETURN
-
-	return verbal, quant
-
-def gre2gmat(gre_verbal,gre_quant):
-	"""
-	:param gre_verbal: Numeric representation of verbal GRE score
-	:type verbal: int
-
-	:param gre_quant: Numeric representation of quantitative GRE score
-	:type quant: int
-
-	:return: Predicted GMAT score
-	:rtype: int
-	"""
-	v,q = _validate_score_input(gre_verbal,gre_quant)
-	if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
-		gmat = BAD_INPUT_RETURN
-	else:
-		try:
-			gmat = score_table.loc[q,v]
-		except KeyError:
-			gmat = rough_gre2gmat(gre_verbal=v,gre_quant=q)
-
-	return gmat
-
-
-def rough_gre2gmat(gre_verbal,gre_quant):
-	"""
-	An equation for computing GMAT equivalent taken from:
-	https://www.mbacrystalball.com/blog/2014/01/23/gre-to-gmat-score-conversion/
-
-	:param gre_verbal: Numeric representation of verbal GRE score
-	:type verbal: int
-
-	:param gre_quant: Numeric representation of quantitative GRE score
-	:type quant: int
-
-	:return: Predicted GMAT score
-	:rtype: int
-	"""
-
-	v,q = _validate_score_input(gre_verbal,gre_quant)
-	if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
-		gmat = BAD_INPUT_RETURN
-	else:
-		gmat = -2080.75 + (6.38 * v) + (10.62 * q)
-		if gmat >= MAX_GMAT:
-			gmat = MAX_GMAT
-		elif gmat <= MIN_GMAT:
-			gmat = MIN_GMAT
-
-	return int(round(gmat,-1))
-def rougher_gre2gmat(gre_verbal,gre_quant):
-	"""
-	A terrible approximation for gmat score. Derived with love by yours truly
-	:param gre_verbal: Numeric representation of verbal GRE score
-	:type verbal: int
-
-	:param gre_quant: Numeric representation of quantitative GRE score
-	:type quant: int
-
-	:return: Predicted GMAT score
-	:rtype: int
-	"""
-	warnings.warn("\nThis function is inappropriately inaccurate.")
-	v,q = _validate_score_input(gre_verbal,gre_quant)
-	if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
-		gmat = BAD_INPUT_RETURN
-	else:
-		gmat = (v+q)*2.019
-
-		if gmat >= MAX_GMAT:
-			gmat = MAX_GMAT
-		elif gmat <= MIN_GMAT:
-			gmat = MIN_GMAT
-
-	return int(round(gmat,-1))
-
-
-
-
-
+from .data_table import score_table
+from .constants import MAX_GRE, MIN_GRE, BAD_INPUT_RETURN, MIN_GMAT, MAX_GMAT
+
+
+def _validate_score_input(v, q):
+    try:
+        verbal = int(v)
+    except:
+        warnings.warn("Was not able to parse verbal score: {}\nReturning {}".format(v, BAD_INPUT_RETURN))
+        return BAD_INPUT_RETURN, BAD_INPUT_RETURN
+    try:
+        quant = int(q)
+    except:
+        warnings.warn("Was not able to parse quant score: {}\nReturning {}".format(q, BAD_INPUT_RETURN))
+        return BAD_INPUT_RETURN, BAD_INPUT_RETURN
+
+    if verbal > MAX_GRE or verbal < MIN_GRE:
+        warnings.warn("Verbal score of {} is not between {} and {}\nReturning {}".format(
+            verbal, MIN_GRE, MAX_GRE, BAD_INPUT_RETURN))
+        return BAD_INPUT_RETURN, BAD_INPUT_RETURN
+
+    if quant > MAX_GRE or quant < MIN_GRE:
+        warnings.warn("Quant score of {} is not between {} and {}\nReturning {}".format(
+            quant, MIN_GRE, MAX_GRE, BAD_INPUT_RETURN))
+        return BAD_INPUT_RETURN, BAD_INPUT_RETURN
+
+    return verbal, quant
+
+
+def gre2gmat(gre_verbal, gre_quant):
+    """
+    :param gre_verbal: Numeric representation of verbal GRE score
+    :type verbal: int
+
+    :param gre_quant: Numeric representation of quantitative GRE score
+    :type quant: int
+
+    :return: Predicted GMAT score
+    :rtype: int
+    """
+    v, q = _validate_score_input(gre_verbal, gre_quant)
+    if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
+        gmat = BAD_INPUT_RETURN
+    else:
+        try:
+            gmat = score_table.loc[q, v]
+        except KeyError:
+            gmat = rough_gre2gmat(gre_verbal=v, gre_quant=q)
+
+    return gmat
+
+
+def rough_gre2gmat(gre_verbal, gre_quant):
+    """
+    An equation for computing GMAT equivalent taken from:
+    https://www.mbacrystalball.com/blog/2014/01/23/gre-to-gmat-score-conversion/
+
+    :param gre_verbal: Numeric representation of verbal GRE score
+    :type verbal: int
+
+    :param gre_quant: Numeric representation of quantitative GRE score
+    :type quant: int
+
+    :return: Predicted GMAT score
+    :rtype: int
+    """
+
+    v, q = _validate_score_input(gre_verbal, gre_quant)
+    if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
+        gmat = BAD_INPUT_RETURN
+    else:
+        gmat = -2080.75 + (6.38 * v) + (10.62 * q)
+        if gmat >= MAX_GMAT:
+            gmat = MAX_GMAT
+        elif gmat <= MIN_GMAT:
+            gmat = MIN_GMAT
+
+    return int(round(gmat, -1))
+
+
+def rougher_gre2gmat(gre_verbal, gre_quant):
+    """
+    A terrible approximation for gmat score. Derived with love by yours truly
+    :param gre_verbal: Numeric representation of verbal GRE score
+    :type verbal: int
+
+    :param gre_quant: Numeric representation of quantitative GRE score
+    :type quant: int
+
+    :return: Predicted GMAT score
+    :rtype: int
+    """
+    warnings.warn("\nThis function is inappropriately inaccurate.")
+    v, q = _validate_score_input(gre_verbal, gre_quant)
+    if v == BAD_INPUT_RETURN or q == BAD_INPUT_RETURN:
+        gmat = BAD_INPUT_RETURN
+    else:
+        gmat = (v + q) * 2.019
+
+        if gmat >= MAX_GMAT:
+            gmat = MAX_GMAT
+        elif gmat <= MIN_GMAT:
+            gmat = MIN_GMAT
+
+    return int(round(gmat, -1))

gre2gmat/data_table.py

@@ -8,9 +8,8 @@
 """
 TABLE_FILE_NAME = 'data_table.csv'
 
-print os.path.dirname(__file__)
-fullpath = os.path.join(os.path.dirname(__file__),TABLE_FILE_NAME)
-score_table = pd.read_csv(fullpath,index_col=0)
+fullpath = os.path.join(os.path.dirname(__file__), TABLE_FILE_NAME)
+score_table = pd.read_csv(fullpath, index_col=0)
 score_table.columns = [int(c) for c in score_table.columns]
 
 """
@@ -24,8 +23,8 @@
 # one40 = [200,220,240,260,280,300,320,340,360,390,410,430,450,470,490,510,530,560,580,600,620]
 # one42 = [210,230,250,270,290,310,330,360,380,400,420,440,460,480,500,530,550,570,590,610,630]
 # one44 = [220,240,260,280,300,330,350,370,390,410,430,450,470,500,520,540,560,580,600,620,640]
-# one46 = [230,250,270,300,320,340,360,380,400,420,440,470,490,510,530,550,570,590,610,640,660]	
-# one48 = [250,270,290,310,330,350,370,390,420,440,460,480,500,520,540,560,580,610,630,650,670]	
+# one46 = [230,250,270,300,320,340,360,380,400,420,440,470,490,510,530,550,570,590,610,640,660]
+# one48 = [250,270,290,310,330,350,370,390,420,440,460,480,500,520,540,560,580,610,630,650,670]
 # one50 = [260,280,300,320,340,360,390,410,430,450,470,490,510,530,560,580,600,620,640,660,680]
 # one52 = [270,290,310,330,360,380,400,420,440,460,480,500,530,550,570,590,610,630,650,670,700]
 # one54 = [280,300,330,350,370,390,410,430,450,470,500,520,540,560,580,600,620,640,670,690,710]