base conversion fix

mathgenerator/funcs/base_conversion.py

@@ -0,0 +1,50 @@
+from .__init__ import *
+
+# base from 2 to 36
+alpha = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+
+
+def fromBaseTenTo(n, toBase):
+    assert type(toBase) == int and toBase >= 2 and toBase <= 36, "toBase({}) must be >=2 and <=36"
+    # trivial cases
+    if toBase == 2:
+        return bin(n)[2:]
+    elif toBase == 8:
+        return oct(n)[2:]
+    elif toBase == 10:
+        return str(n)
+    elif toBase == 16:
+        return hex(n)[2:].upper()
+    res = alpha[n % toBase]
+    n = n//toBase
+    while n > 0:
+        res = alpha[n % toBase] + res
+        n = n//toBase
+    return res
+
+# Useful to check answers, but not needed here
+# def toBaseTen(n,fromBase):
+#     return int(n,fromBase)
+
+
+def baseConversionFunc(maxNum=60000, maxBase=16):
+    assert type(
+        maxNum) == int and maxNum >= 100 and maxNum <= 65536, "maxNum({}) must be >=100 and <=65536".format(maxNum)
+    assert type(
+        maxBase) == int and maxBase >= 2 and maxBase <= 36, "maxBase({}) must be >= 2 and <=36".format(maxBase)
+
+    n = random.randint(40, maxNum)
+    dist = [10]*10+[2]*5+[16]*5+[i for i in range(2, maxBase+1)]
+    # set this way since converting to/from bases 2,10,16 are more common -- can be changed if needed.
+    bases = random.choices(dist, k=2)
+    while bases[0] == bases[1]:
+        bases = random.choices(dist, k=2)
+
+    problem = "Convert {} from base {} to base {}.".format(
+        fromBaseTenTo(n, bases[0]), bases[0], bases[1])
+    ans = fromBaseTenTo(n, bases[1])
+    return problem, ans
+
+
+base_conversion = Generator("Base Conversion", 94, "Convert 152346 from base 8 to base 10.", "54502",
+                            baseConversionFunc)