Initial commitv3.3.4

1 files changed, 73 insertions, 0 deletions

diff --git a/tests/incite/mergers/foundations/test_user_id_product.py b/tests/incite/mergers/foundations/test_user_id_product.py
new file mode 100644
index 0000000..f96bfb4
--- /dev/null
+++ b/tests/incite/mergers/foundations/test_user_id_product.py
@@ -0,0 +1,73 @@
+from datetime import timedelta, datetime, timezone
+from itertools import product
+
+import pandas as pd
+import pytest
+
+# noinspection PyUnresolvedReferences
+from distributed.utils_test import (
+    gen_cluster,
+    client_no_amm,
+    loop,
+    loop_in_thread,
+    cleanup,
+    cluster_fixture,
+    client,
+)
+
+from generalresearch.incite.mergers.foundations.user_id_product import (
+    UserIdProductMergeItem,
+)
+from test_utils.incite.mergers.conftest import user_id_product_merge
+
+
+@pytest.mark.parametrize(
+    argnames="offset, duration, start",
+    argvalues=list(
+        product(
+            ["12h", "3D"],
+            [timedelta(days=5)],
+            [
+                (datetime.now(tz=timezone.utc) - timedelta(days=35)).replace(
+                    microsecond=0
+                )
+            ],
+        )
+    ),
+)
+class TestUserIDProduct:
+
+    @pytest.mark.skip
+    def test_base(self, client_no_amm, user_id_product_merge):
+        ddf = user_id_product_merge.ddf()
+        df = client_no_amm.compute(collections=ddf, sync=True)
+        assert isinstance(df, pd.DataFrame)
+        assert not df.empty
+
+    @pytest.mark.skip
+    def test_base_item(self, client_no_amm, user_id_product_merge, user_collection):
+        assert len(user_id_product_merge.items) == 1
+
+        for item in user_id_product_merge.items:
+            assert isinstance(item, UserIdProductMergeItem)
+
+            path = item.path
+
+            try:
+                modified_time1 = path.stat().st_mtime
+            except (Exception,):
+                modified_time1 = 0
+
+            user_id_product_merge.build(client=client_no_amm, user_coll=user_collection)
+            modified_time2 = path.stat().st_mtime
+
+            assert modified_time2 > modified_time1
+
+    @pytest.mark.skip
+    def test_read(self, client_no_amm, user_id_product_merge):
+        users_ddf = user_id_product_merge.ddf()
+        df = client_no_amm.compute(collections=users_ddf, sync=True)
+
+        assert isinstance(df, pd.DataFrame)
+        assert len(df.columns) == 1
+        assert str(df.product_id.dtype) == "category"