diff --git a/chispa/__init__.py b/chispa/__init__.py index bffde79..d9212c2 100644 --- a/chispa/__init__.py +++ b/chispa/__init__.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import os import sys from glob import glob diff --git a/chispa/bcolors.py b/chispa/bcolors.py index 3c531f2..7b77215 100644 --- a/chispa/bcolors.py +++ b/chispa/bcolors.py @@ -1,3 +1,6 @@ +from __future__ import annotations + + class bcolors: NC = "\033[0m" # No Color, reset all diff --git a/chispa/column_comparer.py b/chispa/column_comparer.py index a3fc43c..e6a380c 100644 --- a/chispa/column_comparer.py +++ b/chispa/column_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from prettytable import PrettyTable from chispa.bcolors import bcolors diff --git a/chispa/dataframe_comparer.py b/chispa/dataframe_comparer.py index 458505b..c88629c 100644 --- a/chispa/dataframe_comparer.py +++ b/chispa/dataframe_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from functools import reduce from chispa.formatting.formats import FormattingConfig diff --git a/chispa/formatting/__init__.py b/chispa/formatting/__init__.py index 0eed329..8983002 100644 --- a/chispa/formatting/__init__.py +++ b/chispa/formatting/__init__.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from chispa.formatting.formats import Color, Format, FormattingConfig, Style from chispa.formatting.terminal_string_formatter import TerminalStringFormatter diff --git a/chispa/formatting/formats.py b/chispa/formatting/formats.py index b6fae78..30ea721 100644 --- a/chispa/formatting/formats.py +++ b/chispa/formatting/formats.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from dataclasses import dataclass, field from enum import Enum diff --git a/chispa/number_helpers.py b/chispa/number_helpers.py index 818d3f5..a49c84a 100644 --- a/chispa/number_helpers.py +++ b/chispa/number_helpers.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import math diff --git a/chispa/row_comparer.py b/chispa/row_comparer.py index a1bf519..7835e5d 100644 --- a/chispa/row_comparer.py +++ b/chispa/row_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import math from pyspark.sql import Row diff --git a/chispa/rows_comparer.py b/chispa/rows_comparer.py index 818b2e6..d8a4ea2 100644 --- a/chispa/rows_comparer.py +++ b/chispa/rows_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from itertools import zip_longest from prettytable import PrettyTable diff --git a/chispa/schema_comparer.py b/chispa/schema_comparer.py index f7cebfc..c2e2150 100644 --- a/chispa/schema_comparer.py +++ b/chispa/schema_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from itertools import zip_longest from prettytable import PrettyTable diff --git a/chispa/structfield_comparer.py b/chispa/structfield_comparer.py index b0fd896..f1b4782 100644 --- a/chispa/structfield_comparer.py +++ b/chispa/structfield_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from chispa.schema_comparer import are_structfields_equal __all__ = ("are_structfields_equal",) diff --git a/docs/gen_ref_pages.py b/docs/gen_ref_pages.py index edd357e..03becb6 100644 --- a/docs/gen_ref_pages.py +++ b/docs/gen_ref_pages.py @@ -3,6 +3,8 @@ https://mkdocstrings.github.io/recipes/#automatic-code-reference-pages """ +from __future__ import annotations + from pathlib import Path import mkdocs_gen_files diff --git a/tests/conftest.py b/tests/conftest.py index 514c2b6..73459d3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from dataclasses import dataclass import pytest diff --git a/tests/formatting/test_formats.py b/tests/formatting/test_formats.py index 4c5cbfb..cd7568c 100644 --- a/tests/formatting/test_formats.py +++ b/tests/formatting/test_formats.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from chispa.formatting.formats import Color, FormattingConfig, Style diff --git a/tests/formatting/test_terminal_string_formatter.py b/tests/formatting/test_terminal_string_formatter.py index 732156c..28cf0f1 100644 --- a/tests/formatting/test_terminal_string_formatter.py +++ b/tests/formatting/test_terminal_string_formatter.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import pytest from chispa.formatting.formats import Color, Format, Style diff --git a/tests/spark.py b/tests/spark.py index c3bd294..b5df699 100644 --- a/tests/spark.py +++ b/tests/spark.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from pyspark.sql import SparkSession spark = SparkSession.builder.master("local").appName("chispa").getOrCreate() diff --git a/tests/test_column_comparer.py b/tests/test_column_comparer.py index 825d3c3..1101545 100644 --- a/tests/test_column_comparer.py +++ b/tests/test_column_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import pytest from chispa import ColumnsNotEqualError, assert_approx_column_equality, assert_column_equality diff --git a/tests/test_dataframe_comparer.py b/tests/test_dataframe_comparer.py index 97035cf..9f14c20 100644 --- a/tests/test_dataframe_comparer.py +++ b/tests/test_dataframe_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import math import pytest diff --git a/tests/test_readme_examples.py b/tests/test_readme_examples.py index bd02ba7..fe28ee0 100644 --- a/tests/test_readme_examples.py +++ b/tests/test_readme_examples.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import pyspark.sql.functions as F import pytest from pyspark.sql import SparkSession diff --git a/tests/test_row_comparer.py b/tests/test_row_comparer.py index a8d13ee..fe9b48c 100644 --- a/tests/test_row_comparer.py +++ b/tests/test_row_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from pyspark.sql import Row from chispa.row_comparer import are_rows_approx_equal, are_rows_equal, are_rows_equal_enhanced diff --git a/tests/test_schema_comparer.py b/tests/test_schema_comparer.py index 0fa87f2..eee7d9b 100644 --- a/tests/test_schema_comparer.py +++ b/tests/test_schema_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import pytest from pyspark.sql.types import ArrayType, DoubleType, IntegerType, StringType, StructField, StructType diff --git a/tests/test_structfield_comparer.py b/tests/test_structfield_comparer.py index 2acf8c7..a6d181d 100644 --- a/tests/test_structfield_comparer.py +++ b/tests/test_structfield_comparer.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from pyspark.sql.types import DoubleType, IntegerType, StructField, StructType from chispa.structfield_comparer import are_structfields_equal