add color table to package data (#124)

* add color table to package data

* added importlib-resources to puproject.toml

---------

Co-authored-by: Petr Andriushchenko <pitandmind@gmail.com>

ammico/test/test_utils.py

@@ -36,3 +36,22 @@ def test_dump_df(get_path):
     df = ut.dump_df(outdict)
     out_df = pd.read_csv(get_path + "example_dump_df.csv", index_col=[0])
     pd.testing.assert_frame_equal(df, out_df)
+
+
+def test_is_interactive():
+    assert ut.is_interactive
+
+
+def test_get_color_table():
+    colors = ut.get_color_table()
+    assert colors["Pink"] == {
+        "ColorName": [
+            "Pink",
+            "LightPink",
+            "HotPink",
+            "DeepPink",
+            "PaleVioletRed",
+            "MediumVioletRed",
+        ],
+        "HEX": ["#FFC0CB", "#FFB6C1", "#FF69B4", "#FF1493", "#DB7093", "#C71585"],
+    }

ammico/utils.py

@@ -2,6 +2,10 @@
 import os
 from pandas import DataFrame, read_csv
 import pooch
+import importlib_resources
+
+
+pkg = importlib_resources.files("ammico")
 
 
 class DownloadResource:
@@ -110,8 +114,9 @@ def is_interactive():
 
 
 def get_color_table():
+    path_tables = pkg / "data" / "Color_tables.csv"
     df_colors = read_csv(
-        os.path.join(os.path.dirname(__file__), "data", "Color_tables.csv"),
+        path_tables,
         delimiter=";",
         dtype=str,
         encoding="UTF-8",

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ammico"
-version = "0.0.1"
+version = "0.1.0"
 description = "AI Media and Misinformation Content Analysis Tool"
 readme = "README.md"
 maintainers = [
@@ -31,6 +31,7 @@ dependencies = [
     "google-cloud-vision",
     "grpcio",
     "importlib_metadata",
+    "importlib_resources",
     "ipython",
     "jupyter_dash",
     "matplotlib",
@@ -71,7 +72,6 @@ documentation = "https://ssciwr.github.io/AMMICO/build/html/index.html"
 packages = ["ammico"]
 
 [tool.setuptools.package-data]
-# Include any png files found in the "data" subdirectory of "ammico"
-"ammico.data" = ["*.png"]
-mypkg = [ "*.csv"]
+# Include any png and csv files found in the "data" subdirectory of "ammico"
+"ammico.data" = ["*.png", "*.csv"]