# HG changeset patch
# User Richard <richardbp@gmail.com>
# Date 1527001843 0
# Node ID 08cfbab35d0b94b6d84a7da971cd4a6571624535
# Parent  06e9b88e0c20c94fd511351185b62cbfc26e64fd# Parent  2fb992c480e465d206e195b38f467ba611a38b01
Merged in kidmose/pywhois/feature/fix-test_parser-keys_to_test (pull request #34)

test_parser.py: Respect keys_to_test/Fix error in 7bec7a730d69

diff -r 06e9b88e0c20 -r 08cfbab35d0b test/test_parser.py
--- a/test/test_parser.py	Tue May 22 11:09:53 2018 -0400
+++ b/test/test_parser.py	Tue May 22 15:10:43 2018 +0000
@@ -85,7 +85,10 @@
                 expected_results = json.load(infil)
 
             # Compare each key
-            for key in set(results).union(set(expected_results)):
+            compare_keys = set.union(set(results), set(expected_results))
+            if keys_to_test is not None:
+                compare_keys = compare_keys.intersection(set(keys_to_test))
+            for key in compare_keys:
                 total += 1
                 if key not in results:
                     print("%s \t(%s):\t Missing in results" % (domain, key,))