Python unittest - potwierdzanie słownika z listami

Question

Podczas pisania testów dla mojej klasy napotkałem interesujący prosty problem. Chciałbym potwierdzić DictEqual dwa słowniki zawierające pewną listę. Ale ten list nie mogą być klasyfikowane w ten sam sposób -> co skutkuje nieudanej próbie

Przykład:

def test_myobject_export_into_dictionary(self): 
    obj = MyObject() 
    resulting_dictionary = { 
      'state': 2347, 
      'neighbours': [1,2,3] 
     } 
    self.assertDictEqual(resulting_dictionary, obj.exportToDict()) 

to nie od czasu do czasu, w zależności od kolejności elementów na liście

FAIL: test_myobject_export_into_dictionary 
------------------------------------ 
- 'neighbours': [1,2,3], 
+ 'neighbours': [1,3,2], 

Jakieś pomysły, jak to potwierdzić w prosty sposób?

Myślałem o używaniu set zamiast list lub sortowania list przed porównaniem.

Answer 1

Można spróbować PyHamcrest(przykład poprawione)

assert_that(obj.exportToDict(), has_entries(
            { 'state': 2347, 
             'neighbours': contains_inanyorder(1,2,3) })) 

(pierwsza wartość 2347 faktycznie zostanie owinięty w sposób dorozumiany equal_to dopasowującego.)

Answer 2

Jak na temat korzystania all:

assert all((k,v) in resulting_dictionary.iteritems() 
      for (k,v) in obj.exportToDict().iteritems()) 

używam coś takiego z py.test, ale myślę, że to powinno działać dla ciebie.

---

commenter wskazał, że zamówienie zostanie wkręcić mnie tutaj --- fair enough ... Chciałbym po prostu korzystać z zestawów, a następnie.

Answer 3

można zrobić:

a = {i:sorted(j) if isinstance(j, list) else j for i,j in resulting_dictionary.iteritems()} 
b = {i:sorted(j) if isinstance(j, list) else j for i,j in obj.exportToDict().iteritems()} 
self.assertDictEqual(a, b) 

Answer 4

może możesz osobno sprawdzić te dwa elementy:

obj_dict = obj.exportToDict() 
    self.assertEqual(resulting_dictionary['state'], obj_dict['state']) 
    self.assertCountEqual(resulting_dictionary['neighbours'], obj_dict['neighbours'])