import unittest

from dungeonsheets.weapons import Weapon


class WeaponTestCase(unittest.TestCase):
    def test_weapon_damage(self):
        weapon = Weapon()
        weapon.base_damage = '1d6'
        self.assertEqual(weapon.damage, '1d6')
        # Now add some bonus damage
        weapon.damage_bonus = 2
        self.assertEqual(weapon.damage, '1d6+2')