Source Code for Module tests.test_single_item_container

 1  #!/usr/bin/env python 
 2   
 3  # This file is part of PARPG. 
 4  #  
 5  # PARPG is free software: you can redistribute it and/or modify 
 6  # it under the terms of the GNU General Public License as published by 
 7  # the Free Software Foundation, either version 3 of the License, or 
 8  # (at your option) any later version. 
 9  #  
10  # PARPG is distributed in the hope that it will be useful, 
11  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
12  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
13  # GNU General Public License for more details. 
14  #  
15  # You should have received a copy of the GNU General Public License 
16  # along with PARPG.  If not, see <http://www.gnu.org/licenses/>. 
17   
18  import unittest 
19  from parpg.objects.base import GameObject, Carryable 
20  from parpg.objects.composed import SingleItemContainer 
21   
22 -class  TestSingleItemContainer(unittest.TestCase): 
23   
24 -    class CarryableObject (GameObject, Carryable): 
25 -        def __init__ (self, ID, **kwargs): 
26              GameObject.__init__(self, ID, **kwargs) 
27              Carryable.__init__(self, **kwargs) 
28   
29   
30 -    def setUp(self): 
31          self.item = self.CarryableObject(6) 
32          self.item2 = self.CarryableObject(7) 
33   
34 -    def tearDown(self): 
35          self.item = None 
36          self.item2 = None 
37   
38   
39 -    def testPlaceTake(self): 
40          """ Test SingleItemContainer Place/Take functions """ 
41          container = SingleItemContainer() 
42          container.placeItem(self.item) 
43          self.assertRaises(container.SlotBusy, container.placeItem, self.item2) 
44          container.takeItem(self.item) 
45          self.assertEqual(container.items, {}) 
46          container.placeItem(self.item2) 
47   
48  if __name__ == '__main__': 
49      unittest.main() 
50