Formatting; Fixing tests

pandas_profiling/__init__.py

@@ -175,7 +175,6 @@ def default(self, o):
 
         return json.dumps(self.description_set, indent=4, cls=CustomEncoder)
 
-
     def _repr_html_(self):
         """Used to output the HTML representation to a Jupyter notebook.
         When config.notebook.iframe.attribute is "src", this function creates a temporary HTML file

pandas_profiling/config.py

@@ -56,7 +56,7 @@ def __getitem__(self, item):
 
     def __setitem__(self, key, value):
         self.config[key].set(value)
-        
+
     def dump(self):
         return self.config.dump()
 

pandas_profiling/model/messages.py

@@ -55,7 +55,10 @@ class Message(object):
     """A message object (type, values, column)."""
 
     def __init__(
-        self, message_type: MessageType, values: dict, column_name: Union[str, None] = None
+        self,
+        message_type: MessageType,
+        values: dict,
+        column_name: Union[str, None] = None,
     ):
         self.message_type = message_type
         self.values = values

tests/unit/test_describe.py

@@ -627,9 +627,14 @@ def test_describe_df(describe_data, expected_results):
 
     results = describe(describe_data_frame)
 
-    assert {"table", "variables", "correlations", "missing", "messages"} == set(
-        results.keys()
-    ), "Not in results"
+    assert {
+        "table",
+        "variables",
+        "correlations",
+        "missing",
+        "messages",
+        "package",
+    } == set(results.keys()), "Not in results"
     assert set(
         {
             "CAT": 1,

tests/unit/test_example.py

@@ -43,7 +43,7 @@ def test_example(tmpdir):
     assert (tmpdir / "profile.html").exists(), "Output file does not exist"
     assert (
         type(profile.get_description()) == dict
-        and len(profile.get_description().items()) == 5
+        and len(profile.get_description().items()) == 6
     ), "Unexpected result"
     assert set(profile.get_rejected_variables(0.9)) == {
         "reclat_city"