Testing

Sometimes, you might want to use the same data to test different functions. If you are using pytest, use pytest fixtures to provide data to different test functions instead.
In the code above, I used the decorator @pytest.fixture to create the data I want to use many times.
Have this data as the input for any test function that uses this data!

It is a good practice to test your functions to make sure they work as expected, but sometimes you need to test 100 times until you found the rare cases when the test fails. That is when pytest-repeat comes in handy.
Yesterday, I tested with 100 instances and found 2 failed tests! I’m glad that I didn’t settle down for 10 passed tests. You can find the doc for pytest-repeat here

Your code might work with data at a certain state but might not work if your data changes. How do you detect the change in data before running into errors or encounter the drop in your model performance? That is when Schema comes in handy.
Below is the code snippet I used to validate the datatypes and the range of the values in a column. If this is what you are looking for, check out my article on how to use Schema here.