How can you test that two dictionaries are equal with pytest in python

Question:

Trying to assert that two dictionaries that have nested contents are equal to each other (order doesn’t matter) with pytest. What’s the pythonic way to do this?

Asked By: John Mike

||

Answers:

I guess a simple assert equality test should be okay:

>>> d1 = {n: chr(n+65) for n in range(10)}
>>> d2 = {n: chr(n+65) for n in range(10)}
>>> d1 == d2
True
>>> l1 = [1, 2, 3]
>>> l2 = [1, 2, 3]
>>> d2[10] = l2
>>> d1[10] = l1
>>> d1 == d2
True
>>> class Example:
    stub_prop = None
>>> e1 = Example()
>>> e2 = Example()
>>> e2.stub_prop = 10
>>> e1.stub_prop = 'a'
>>> d1[11] = e1
>>> d2[11] = e2
>>> d1 == d2
False
Answered By: Szabolcs

your question is not very specific but with what i can understand, you are either trying to check if the length are the same

a = [1,5,3,6,3,2,4]
b = [5,3,2,1,3,5,3]

if (len(a) == len(b)):
    print True
else:
    print false

or checking if the list values are the same

import collections

compare = lambda x, y: collections.Counter(x) == collections.Counter(y)
compare([1,2,3], [1,2,3,3])
print compare #answer would be false
compare([1,2,3], [1,2,3])
print compare #answer would be true

but for dictionaries you could also use

x = dict(a=1, b=2)
y = dict(a=2, b=2)

if(x == y):
    print True
else:
    print False
Answered By: Adeojo Emmanuel IMM
assert all(v == actual_dict[k] for k,v expected_dict.items()) and len(expected_dict) == len(actual_dict)

General purpose way is to:

import json

# Make sure you sort any lists in the dictionary before dumping to a string

dictA_str = json.dumps(dictA, sort_keys=True)
dictB_str = json.dumps(dictB, sort_keys=True)

assert dictA_str == dictB_str
Answered By: William

Don’t spend your time writing this logic yourself. Just use the functions provided by the default testing library unittest

from unittest import TestCase
TestCase().assertDictEqual(expected_dict, actual_dict)
Answered By: Vincent Claes

pytest’s magic is clever enough. By writing

assert {'a': {'b': 2, 'c': {'d': 4} } } == {'a': {'b': 2, 'c': {'d': 4} } }

you will have a nested test on equality.

Answered By: linqu
Categories: questions Tags: , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.