test_enchanting.py 1.2 KB

123456789101112131415161718192021222324252627
  1. import enchanting
  2. def test_volatility_from_strength():
  3. assert(enchanting._volatility_from_spell_level(3) == 1)
  4. assert(enchanting._volatility_from_spell_level(5) == 1)
  5. assert(enchanting._volatility_from_spell_level(6) == 2)
  6. assert(enchanting._volatility_from_spell_level(9) == 3)
  7. def test_volatility_from_time():
  8. assert(enchanting._volatility_from_time(enchanting.TimeFactor.RUSHING) == 1)
  9. assert(enchanting._volatility_from_time(enchanting.TimeFactor.RUSHING_EVEN_MORE) == 2)
  10. assert(enchanting._volatility_from_time(enchanting.TimeFactor.PAITENCE) == -1)
  11. assert(enchanting._volatility_from_time(enchanting.TimeFactor.EVEN_MORE_PAITENCE) == -2)
  12. assert(enchanting._volatility_from_time(enchanting.TimeFactor.NORMAL) == 0)
  13. def test_volatility():
  14. first = enchanting.calculate_volatility(enchanting.EnchantmentType.DAILY_CHARGES,
  15. 1,
  16. enchanting.TimeFactor.RUSHING_EVEN_MORE, 1, 0)
  17. assert(first == 4)
  18. second = enchanting.calculate_volatility(enchanting.EnchantmentType.DAILY_CHARGES,
  19. 2,
  20. enchanting.TimeFactor.RUSHING_EVEN_MORE, 5, 0)