ENH: improved node consolidation functions

ci/envs/38-minimal.yaml

@@ -5,7 +5,7 @@ dependencies:
   - python=3.8
   - geopandas=0.8
   - libpysal=4.2.2
-  - networkx=2.3
+  - networkx=2.5
   - tqdm=4.27.0
   - pytest
   - pytest-cov

momepy/datasets/__init__.py

@@ -3,20 +3,23 @@
 __all__ = ["available", "get_path"]
 
 _module_path = os.path.dirname(__file__)
-available = ["bubenec", "tests"]
+available = ["bubenec", "tests", "nyc_graph"]
 
 
-def get_path(dataset):
+def get_path(dataset, extension="gpkg"):
     """
     Get the path to the data file.
     Parameters
     ----------
     dataset : str
         The name of the dataset. See ``momepy.datasets.available`` for
         all options.
+    extension : str
+        The extension of the data file
     """
     if dataset in available:
-        return os.path.abspath(os.path.join(_module_path, dataset + ".gpkg"))
+        filepath = dataset + "." + extension
+        return os.path.abspath(os.path.join(_module_path, filepath))
     msg = "The dataset '{data}' is not available. ".format(data=dataset)
     msg += "Available datasets are {}".format(", ".join(available))
     raise ValueError(msg)