Fix test in Python 2.7

datarobot · Aug 29, 2017 · f7d375a · f7d375a
1 parent f55f51a
commit f7d375a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_api_response_handlers.py b/tests/test_api_response_handlers.py
@@ -140,7 +140,7 @@ def test_unpack_data(self):
         assert type(result) is tuple
         data, exec_time, elapsed = result
         assert type(data) is dict
-        assert data.keys() == {
+        assert set(data.keys()) == {
             'status', 'model_id', 'code', 'execution_time', 'predictions'}
         assert elapsed == 0.123
         assert exec_time == 10