aboutsummaryrefslogtreecommitdiff
path: root/tests/incite/collections
diff options
context:
space:
mode:
authorMax Nanis2026-08-24 12:35:35 -0700
committerMax Nanis2026-08-24 12:35:35 -0700
commite2c5de703be45746bacaea4136f24440ff5a291c (patch)
tree9c3d9d448e56357feda24a4b0749370a9a33cb55 /tests/incite/collections
parentd48f04ab7f034b9b2039c42cd616d7d7851817b9 (diff)
downloadgeneralresearch-e2c5de703be45746bacaea4136f24440ff5a291c.tar.gz
generalresearch-e2c5de703be45746bacaea4136f24440ff5a291c.zip
Ruff std replacements
Diffstat (limited to 'tests/incite/collections')
-rw-r--r--tests/incite/collections/test_df_collection_base.py3
-rw-r--r--tests/incite/collections/test_df_collection_item_base.py2
-rw-r--r--tests/incite/collections/test_df_collection_item_thl_web.py11
-rw-r--r--tests/incite/collections/test_df_collection_thl_marketplaces.py7
-rw-r--r--tests/incite/collections/test_df_collection_thl_web.py2
5 files changed, 4 insertions, 21 deletions
diff --git a/tests/incite/collections/test_df_collection_base.py b/tests/incite/collections/test_df_collection_base.py
index c3c64e4..176bf4b 100644
--- a/tests/incite/collections/test_df_collection_base.py
+++ b/tests/incite/collections/test_df_collection_base.py
@@ -1,4 +1,4 @@
-from datetime import UTC, datetime, timezone
+from datetime import UTC, datetime
from typing import TYPE_CHECKING
import pandas as pd
@@ -9,7 +9,6 @@ from generalresearch.incite.collections import (
DFCollection,
DFCollectionType,
)
-from test_utils.incite.conftest import mnt_filepath
if TYPE_CHECKING:
from generalresearch.incite.base import GRLDatasets
diff --git a/tests/incite/collections/test_df_collection_item_base.py b/tests/incite/collections/test_df_collection_item_base.py
index 8cf719d..0218f30 100644
--- a/tests/incite/collections/test_df_collection_item_base.py
+++ b/tests/incite/collections/test_df_collection_item_base.py
@@ -1,4 +1,4 @@
-from datetime import UTC, datetime, timezone
+from datetime import UTC, datetime
from typing import TYPE_CHECKING
import pytest
diff --git a/tests/incite/collections/test_df_collection_item_thl_web.py b/tests/incite/collections/test_df_collection_item_thl_web.py
index 062171d..3d70e56 100644
--- a/tests/incite/collections/test_df_collection_item_thl_web.py
+++ b/tests/incite/collections/test_df_collection_item_thl_web.py
@@ -1,7 +1,7 @@
from __future__ import annotations
from collections.abc import Callable, Generator
-from datetime import UTC, datetime, timedelta, timezone
+from datetime import UTC, datetime, timedelta
from itertools import product as iter_product
from os.path import join as pjoin
from pathlib import Path, PurePath
@@ -236,7 +236,6 @@ class TestDFCollectionItemMethod:
incite_item_factory,
delete_df_collection,
):
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -280,7 +279,6 @@ class TestDFCollectionItemMethod:
incite_item_factory,
delete_df_collection,
):
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -333,7 +331,6 @@ class TestDFCollectionItemMethod:
delete_df_collection,
mnt_filepath,
):
- from generalresearch.models.thl.user import User
if df_collection.data_type != DFCollectionType.LEDGER:
return
@@ -385,7 +382,6 @@ class TestDFCollectionItemMethod:
delete_df_collection,
mnt_filepath,
):
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -766,7 +762,6 @@ class TestDFCollectionItemFunctionalTest:
delete_df_collection,
mnt_filepath: GRLDatasets,
):
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -828,7 +823,6 @@ class TestDFCollectionItemFunctionalTest:
import pyarrow.parquet as pq
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -864,7 +858,6 @@ class TestDFCollectionItemFunctionalTest:
delete_df_collection,
mnt_filepath: GRLDatasets,
):
- from generalresearch.models.thl.user import User
delete_df_collection(coll=df_collection)
df_collection._client = client_no_amm
@@ -920,7 +913,6 @@ class TestDFCollectionItemFunctionalTest:
"""Don't allow creating an archive for data that will likely be
overwritten or updated
"""
- from generalresearch.models.thl.user import User
if df_collection.data_type in unsupported_mock_types:
return
@@ -987,7 +979,6 @@ class TestDFCollectionItemFunctionalTest:
duration: timedelta,
mnt_filepath,
):
- from generalresearch.models.thl.user import User
delete_df_collection(coll=df_collection)
df_collection._client = client_no_amm
diff --git a/tests/incite/collections/test_df_collection_thl_marketplaces.py b/tests/incite/collections/test_df_collection_thl_marketplaces.py
index 2597d38..d2d3ce4 100644
--- a/tests/incite/collections/test_df_collection_thl_marketplaces.py
+++ b/tests/incite/collections/test_df_collection_thl_marketplaces.py
@@ -1,6 +1,5 @@
-from datetime import UTC, datetime, timezone
+from datetime import UTC, datetime
from itertools import product
-from typing import TYPE_CHECKING
import pytest
from pandera.pandas import Column, DataFrameSchema, Index
@@ -12,10 +11,6 @@ from generalresearch.incite.collections.thl_marketplaces import (
SagoSurveyHistoryCollection,
SpectrumSurveyTimeseriesCollection,
)
-from test_utils.incite.conftest import mnt_filepath
-
-if TYPE_CHECKING:
- from generalresearch.incite.base import GRLDatasets
def combo_object():
diff --git a/tests/incite/collections/test_df_collection_thl_web.py b/tests/incite/collections/test_df_collection_thl_web.py
index 2cb0ba0..bcdeb83 100644
--- a/tests/incite/collections/test_df_collection_thl_web.py
+++ b/tests/incite/collections/test_df_collection_thl_web.py
@@ -13,9 +13,7 @@ from pandera.pandas import DataFrameSchema
from generalresearch.incite.collections import DFCollection, DFCollectionType
if TYPE_CHECKING:
- from generalresearch.incite.base import GRLDatasets
from generalresearch.incite.collections import (
- DFCollectionItem,
DFCollectionType,
)