Python对象测试

Question

我有以下单元测试问题，我试图通过。

def test_map2(self): 
    self.home = [] 
    self.home.append(Bed('Bedroom')) 
    self.home.append(Sofa('Living Room')) 
    self.home.append(Table('Bedroom')) 

    mapping = map_the_home(self.home) 
    self.assertTrue(isinstance(mapping['Bedroom'][0], Bed)) 
    self.assertTrue(isinstance(mapping['Living Room'][0], Sofa)) 
    self.assertTrue(isinstance(mapping['Bedroom'][1], Table)) 

每个值都应该是一个列表，里面有一个或多个Furnishing子类实例。

这是我目前的尝试。

class Furnishing(object):  
    def __init__(self, room): 
     self.room = room 

class Sofa(Furnishing):  
    name = 'Sofa' 

class Bed(Furnishing):  
    name = 'Bed' 

class Table(Furnishing):  
    name = 'Table' 

def map_the_home(home):  
    results = {} 

    for furnitiure in home:  
     if furnitiure.room in results:     
      results[furnitiure.room] = (results[furnitiure.room],furnitiure)  
     else:  
      results[furnitiure.room] = furnitiure   
    return results 

def counter(home):  
    counter_list = {} 
    for line in home:  
     if line.name in counter_list: 
      print(line.room,line.name) 
      counter_list[line.name] = counter_list[line.name] + 1  
     else:  
      counter_list[line.name] = 1  

    for furniture in counter_list:  
     print('{0} = {1}'.format(furniture,counter_list[furniture])) 

if __name__ == "__main__":  
    home = [] 
    home.append(Bed('Bedroom')) 
    home.append(Sofa('Living Room')) 
    home.append(Table('Bedroom')) 
    map_the_home(home) 
    counter(home) 

该计数器只是另一部分，但希望给出完整的代码。我以为我有这个使用字典，但作为测试说我需要在每个值中的家具子类实例内的列表。任何有识之士将是巨大的

Answer 1

测试期待，其结果将是这样的：

mapping == {'Bedroom': [bed, table], 'Living Room': [sofa]} 

而创建：

{'Bedroom': (bed, table), 'Living Room': sofa} 

备注"Living Room"的值不是容器，而是单个的Sofa实例。

事实上，如果你在一个房间里有三件家具，例如在“卧室”，并称一个Sofa：

{'Bedroom': ((bed, table), sofa), 'Living Room': sofa} 

你会继续嵌套更深。

最小的解决方法是：

if furniture.room in results:     
    results[furniture.room].append(furniture)  
else:  
    results[furniture.room] = [furniture] 

注意使用list和tuple可以得到同样的结果，既可以被索引（虽然你add（不append）到元组）;我认为这里列表更适合，但是你使用元组不是错误的来源。

Answer 2

def map_the_home(home): 
    results = dict() 
    for furniture in home: 
     results.setdefault(furniture.room,[]).append(furniture) 
    return results 

让我们通过它的伪

home = [Bed('Bedroom'), Sofa('Living Room'), Table('Bedroom')] 
map_the_home(home) 
results = dict() 
for each piece of furniture in the home: 
    append that piece of furniture to results[furniture.room] 
    if results[furniture.room] doesn't exist, make it an empty list then append 
return the results