Merge branch 'master' into jvandooren/rule-configuration

pyproject.toml

@@ -55,13 +55,13 @@ dbt-score = {call = "dbt_score.cli:cli"}
 
 [tool.mypy]
 strict = true
+warn_no_return = false
 
 [[tool.mypy.overrides]]
 module = "tests.*"
 disallow_untyped_defs = false
 disallow_incomplete_defs = false
 disallow_untyped_calls = false
-warn_no_return = false
 
 ### Ruff ###
 

src/dbt_score/__init__.py

@@ -1 +1,6 @@
 """Init dbt_score package."""
+
+from dbt_score.models import Model
+from dbt_score.rule import Rule, RuleViolation, Severity, rule
+
+__all__ = ["Model", "Rule", "RuleViolation", "Severity", "rule"]

src/dbt_score/rules/generic.py

@@ -1,9 +1,6 @@
 """All generic rules."""
 
-from dbt_score.models import Model
-from dbt_score.rule import RuleViolation, rule
-
-# mypy: disable-error-code="return"
+from dbt_score import Model, RuleViolation, rule
 
 
 @rule

tests/conftest.py

@@ -5,8 +5,7 @@
 from typing import Any, Type
 
 from dbt_score.config_parser import DbtScoreConfig, RuleConfig
-from dbt_score.models import Model
-from dbt_score.rule import Rule, RuleViolation, Severity, rule
+from dbt_score import Model, Rule, RuleViolation, Severity, rule
 from pytest import fixture
 
 # Configuration

tests/rules/example.py

@@ -1,7 +1,6 @@
 """Example rules."""
 
-from dbt_score.models import Model
-from dbt_score.rule import RuleViolation, rule
+from dbt_score import Model, RuleViolation, rule
 
 
 @rule()

tests/rules/nested/example.py

@@ -1,9 +1,8 @@
 """Example rules."""
 
-from dbt_score.models import Model
-from dbt_score.rule import RuleViolation, rule
+from dbt_score import Model, RuleViolation, rule
 
 
-@rule()
+@rule
 def rule_test_nested_example(model: Model) -> RuleViolation | None:
     """An example rule."""

tests/test_rule.py

@@ -1,8 +1,7 @@
 """Test rule."""
 
 import pytest
-from dbt_score.models import Model
-from dbt_score.rule import Rule, RuleViolation, Severity, rule
+from dbt_score import Model, Rule, RuleViolation, Severity, rule
 
 
 def test_rule_decorator_and_class(