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.bonus_damage = 2
        self.assertEqual(weapon.damage, '1d6 +2')
