Skip to content

Commit f35bbd9

Browse files
committed
changed all Dataframe references to DataFrame
1 parent 09c712d commit f35bbd9

File tree

57 files changed

+1692
-1689
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

57 files changed

+1692
-1689
lines changed

README.md

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,21 +24,24 @@ cd dataframe-expectations
2424
# 2. Install UV package manager
2525
pip install uv
2626

27-
# 3. Install development dependencies
27+
# 3. Install development dependencies (this will automatically create a virtual environment)
2828
uv sync --group dev
2929

30-
# 4. Run tests
30+
# 4. (Optional) To explicitly activate the virtual environment:
31+
source .venv/bin/activate # On Windows: .venv\Scripts\activate
32+
33+
# 5. Run tests (this will run the tests in the virtual environment)
3134
uv run pytest tests/ --cov=dataframe_expectations
3235
```
3336

3437
### Using the library
3538

3639
**Pandas example:**
3740
```python
38-
from dataframe_expectations.expectations_suite import DataframeExpectationsSuite
41+
from dataframe_expectations.expectations_suite import DataFameExpectationsSuite
3942

4043
suite = (
41-
DataframeExpectationsSuite()
44+
DataFrameExpectationsSuite()
4245
.expect_value_greater_than("age", 18)
4346
.expect_value_less_than("age", 10)
4447
)
@@ -54,10 +57,10 @@ suite.run(test_pandas_df)
5457

5558
**PySpark example:**
5659
```python
57-
from dataframe_expectations.expectations_suite import DataframeExpectationsSuite
60+
from dataframe_expectations.expectations_suite import DataFrameExpectationsSuite
5861

5962
suite = (
60-
DataframeExpectationsSuite()
63+
DataFrameExpectationsSuite()
6164
.expect_value_greater_than("age", 18)
6265
.expect_value_less_than("age", 40)
6366
)

dataframe_expectations/expectations/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,11 @@
1313

1414
from dataframe_expectations import DataFrameLike, DataFrameType
1515
from dataframe_expectations.result_message import (
16-
DataframeExpectationResultMessage,
16+
DataFrameExpectationResultMessage,
1717
)
1818

1919

20-
class DataframeExpectation(ABC):
20+
class DataFrameExpectation(ABC):
2121
"""
2222
Base class for DataFrame expectations.
2323
"""
@@ -75,7 +75,7 @@ def validate(self, data_frame: DataFrameLike, **kwargs):
7575
@abstractmethod
7676
def validate_pandas(
7777
self, data_frame: DataFrameLike, **kwargs
78-
) -> DataframeExpectationResultMessage:
78+
) -> DataFrameExpectationResultMessage:
7979
"""
8080
Validate a pandas DataFrame against the expectation.
8181
"""
@@ -86,7 +86,7 @@ def validate_pandas(
8686
@abstractmethod
8787
def validate_pyspark(
8888
self, data_frame: DataFrameLike, **kwargs
89-
) -> DataframeExpectationResultMessage:
89+
) -> DataFrameExpectationResultMessage:
9090
"""
9191
Validate a PySpark DataFrame against the expectation.
9292
"""

dataframe_expectations/expectations/aggregation_expectation.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,14 @@
22
from typing import List, Union
33

44
from dataframe_expectations import DataFrameLike, DataFrameType
5-
from dataframe_expectations.expectations import DataframeExpectation
5+
from dataframe_expectations.expectations import DataFrameExpectation
66
from dataframe_expectations.result_message import (
7-
DataframeExpectationFailureMessage,
8-
DataframeExpectationResultMessage,
7+
DataFrameExpectationFailureMessage,
8+
DataFrameExpectationResultMessage,
99
)
1010

1111

12-
class DataframeAggregationExpectation(DataframeExpectation):
12+
class DataFrameAggregationExpectation(DataFrameExpectation):
1313
"""
1414
Base class for DataFrame aggregation expectations.
1515
This class is designed to first aggregate data and then validate the aggregation results.
@@ -48,7 +48,7 @@ def get_description(self) -> str:
4848
@abstractmethod
4949
def aggregate_and_validate_pandas(
5050
self, data_frame: DataFrameLike, **kwargs
51-
) -> DataframeExpectationResultMessage:
51+
) -> DataFrameExpectationResultMessage:
5252
"""
5353
Aggregate and validate a pandas DataFrame against the expectation.
5454
@@ -62,7 +62,7 @@ def aggregate_and_validate_pandas(
6262
@abstractmethod
6363
def aggregate_and_validate_pyspark(
6464
self, data_frame: DataFrameLike, **kwargs
65-
) -> DataframeExpectationResultMessage:
65+
) -> DataFrameExpectationResultMessage:
6666
"""
6767
Aggregate and validate a PySpark DataFrame against the expectation.
6868
@@ -75,15 +75,15 @@ def aggregate_and_validate_pyspark(
7575

7676
def validate_pandas(
7777
self, data_frame: DataFrameLike, **kwargs
78-
) -> DataframeExpectationResultMessage:
78+
) -> DataFrameExpectationResultMessage:
7979
"""
8080
Validate a pandas DataFrame against the expectation.
8181
Automatically checks column existence before calling the implementation.
8282
"""
8383
# Check if all required columns exist
8484
column_error = self._check_columns_exist(data_frame)
8585
if column_error:
86-
return DataframeExpectationFailureMessage(
86+
return DataFrameExpectationFailureMessage(
8787
expectation_str=str(self),
8888
data_frame_type=DataFrameType.PANDAS,
8989
message=column_error,
@@ -94,15 +94,15 @@ def validate_pandas(
9494

9595
def validate_pyspark(
9696
self, data_frame: DataFrameLike, **kwargs
97-
) -> DataframeExpectationResultMessage:
97+
) -> DataFrameExpectationResultMessage:
9898
"""
9999
Validate a PySpark DataFrame against the expectation.
100100
Automatically checks column existence before calling the implementation.
101101
"""
102102
# Check if all required columns exist
103103
column_error = self._check_columns_exist(data_frame)
104104
if column_error:
105-
return DataframeExpectationFailureMessage(
105+
return DataFrameExpectationFailureMessage(
106106
expectation_str=str(self),
107107
data_frame_type=DataFrameType.PYSPARK,
108108
message=column_error,

0 commit comments

Comments
 (0)