Parsing the grammar generates a rollable dice

Author: Bernardo-MG

dice_notation/parser/DiceNotationListener.py

@@ -2,7 +2,7 @@
 from antlr4 import *
 
 import logging
-from dice_notation.dice import Dice
+from dice_notation.dice import RollableDice
 from dice_notation.algebra import BinaryOperation, Number
 
 # This class defines a complete listener for a parse tree produced by DiceNotationParser.
@@ -125,7 +125,7 @@ def exitDice(self, ctx):
 
         sides = int(next(digits).getText())
 
-        dice = Dice(quantity, sides)
+        dice = RollableDice(quantity, sides)
         self._nodes.append(dice)
 
 

tests/parser/dice/test_dice_parser_dice_roll.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+
+import sys
+import unittest
+
+from dice_notation.parser import DiceParser
+
+"""
+Dice parser tests for expressions only containing dice.
+"""
+
+__author__ = 'Bernardo Martínez Garrido'
+__license__ = 'MIT'
+
+
+class TestSimpleDice(unittest.TestCase):
+    """
+    Tests that simple dice expressions can be parsed.
+    """
+
+    def setUp(self):
+        """
+        Initializes parser.
+        """
+        self.parser = DiceParser()
+
+    def test_minimalDice(self):
+        """
+        Tests that the minimal dice can be rolled.
+        """
+        result = self.parser.parse("1d1").roll()
+
+        self.assertEqual(1, result)