docs: simplify comment styles
This commit is contained in:
@@ -27,9 +27,7 @@ ALL_JOBS = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Generic schedule tests
|
# Generic schedule tests
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestSchedulesGeneric:
|
class TestSchedulesGeneric:
|
||||||
@@ -53,9 +51,7 @@ class TestSchedulesGeneric:
|
|||||||
assert schedule.default_status == DefaultScheduleStatus.RUNNING
|
assert schedule.default_status == DefaultScheduleStatus.RUNNING
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Generic job tests
|
# Generic job tests
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestJobsGeneric:
|
class TestJobsGeneric:
|
||||||
@@ -70,9 +66,7 @@ class TestJobsGeneric:
|
|||||||
assert job.description
|
assert job.description
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Schedule-specific tests
|
# Schedule-specific tests
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestScheduleSpecific:
|
class TestScheduleSpecific:
|
||||||
@@ -89,9 +83,7 @@ class TestScheduleSpecific:
|
|||||||
assert funda_ingestion_schedule._run_config_fn is not None
|
assert funda_ingestion_schedule._run_config_fn is not None
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Definitions integration test
|
# Definitions integration test
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestDefinitions:
|
class TestDefinitions:
|
||||||
|
|||||||
@@ -21,9 +21,7 @@ from data_platform.assets.ml.discord_alerts import (
|
|||||||
from data_platform.assets.ml.elo_inference import EloInferenceConfig, _best_run
|
from data_platform.assets.ml.elo_inference import EloInferenceConfig, _best_run
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# preprocess
|
# preprocess
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestPreprocess:
|
class TestPreprocess:
|
||||||
@@ -101,9 +99,7 @@ class TestPreprocess:
|
|||||||
assert feat in df.columns, f"Missing feature: {feat}"
|
assert feat in df.columns, f"Missing feature: {feat}"
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# _best_run
|
# _best_run
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestBestRun:
|
class TestBestRun:
|
||||||
@@ -152,9 +148,7 @@ class TestBestRun:
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# _build_embed
|
# _build_embed
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestBuildEmbed:
|
class TestBuildEmbed:
|
||||||
@@ -209,9 +203,7 @@ class TestBuildEmbed:
|
|||||||
assert embed["color"] == 0x00B894
|
assert embed["color"] == 0x00B894
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# _cleanup_old_runs
|
# _cleanup_old_runs
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestCleanupOldRuns:
|
class TestCleanupOldRuns:
|
||||||
@@ -257,9 +249,7 @@ class TestCleanupOldRuns:
|
|||||||
client.delete_run.assert_not_called()
|
client.delete_run.assert_not_called()
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# elo_prediction_model asset
|
# elo_prediction_model asset
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
def _make_training_df(n=20):
|
def _make_training_df(n=20):
|
||||||
@@ -354,9 +344,7 @@ class TestEloPredictionModelAsset:
|
|||||||
elo_prediction_model(context, config, postgres, mlflow_resource)
|
elo_prediction_model(context, config, postgres, mlflow_resource)
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# elo_inference asset
|
# elo_inference asset
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestEloInferenceAsset:
|
class TestEloInferenceAsset:
|
||||||
@@ -457,9 +445,7 @@ class TestEloInferenceAsset:
|
|||||||
assert result.metadata["status"].value == "No new listings to score."
|
assert result.metadata["status"].value == "No new listings to score."
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# listing_alert asset
|
# listing_alert asset
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestListingAlertAsset:
|
class TestListingAlertAsset:
|
||||||
@@ -570,9 +556,7 @@ class TestListingAlertAsset:
|
|||||||
assert mock_post.call_count == 2
|
assert mock_post.call_count == 2
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Config defaults
|
# Config defaults
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestMLConfigs:
|
class TestMLConfigs:
|
||||||
|
|||||||
@@ -16,9 +16,7 @@ from data_platform.ops.elementary import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# SourceFreshnessConfig
|
# SourceFreshnessConfig
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestSourceFreshnessConfig:
|
class TestSourceFreshnessConfig:
|
||||||
@@ -27,9 +25,7 @@ class TestSourceFreshnessConfig:
|
|||||||
assert cfg.source_name == "raw_funda"
|
assert cfg.source_name == "raw_funda"
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# _elementary_schema_exists
|
# _elementary_schema_exists
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestElementarySchemaExists:
|
class TestElementarySchemaExists:
|
||||||
@@ -69,9 +65,7 @@ class TestElementarySchemaExists:
|
|||||||
assert result is False
|
assert result is False
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# elementary_run_models
|
# elementary_run_models
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestElementaryRunModels:
|
class TestElementaryRunModels:
|
||||||
@@ -110,9 +104,7 @@ class TestElementaryRunModels:
|
|||||||
elementary_run_models(context)
|
elementary_run_models(context)
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# elementary_generate_report
|
# elementary_generate_report
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestElementaryGenerateReport:
|
class TestElementaryGenerateReport:
|
||||||
|
|||||||
Reference in New Issue
Block a user