def test_fill_nulls_with_zeros(): dataset = Dataset('src/data/Health_data.csv') dataset.load_data() preprocess = Preprocess(dataset.get_data()) # First preprocess object with filled nulls preprocess = preprocess.fill_nulls_with_zeros() # Another preprocess object that has filled nulls zeros = Preprocess(dataset.get_data()).fill_nulls_with_zeros() assert preprocess == zeros
def test_plot_nulls_in_the_eda(): dataset = Dataset('src/data/Health_data.csv') dataset.load_data() preprocess = Preprocess(dataset.get_data()) eda = EDA(preprocess.get_preprocessed_so_far()).plot_nulls() eda_nulls = EDA(preprocess.get_preprocessed_so_far()).plot_nulls() assert eda == eda_nulls
def test_preprocess_delegation_to_eda_object(): '''EDA gets passed the data that has undergone preprocessing. In the future, I can refactor to make Preprocess immutable objects. So far EDA is immutable.''' dataset = Dataset('src/data/Health_data.csv') dataset.load_data() preprocess = Preprocess(dataset.get_data()) eda = EDA(preprocess.get_preprocessed_so_far()) assert eda
def test_delegation_to_preprocess_object(): dataset = Dataset('src/data/Health_data.csv') preprocess = Preprocess(dataset.get_data()) assert preprocess