#!/usr/bin/env python

import unittest

import math

from pyutil.mathutil import *
from pyutil.assertutil import _assert

class MathUtilTestCase(unittest.TestCase):
    def _help_test_is_power_of_k(self, k):
        for i in range(2, 40):
            _assert(is_power_of_k(k**i, k), k, i)

    def test_is_power_of_k(self):
        for i in range(2, 5):
            self._help_test_is_power_of_k(i)

def suite():
    return unittest.makeSuite(MathUtilTestCase, 'test')

if __name__ == "__main__":
    unittest.main()