Python impact_coding_on_a_single_column Examples

Example #1

File: test_impact_coding.py Project: AbhishekRatnawat/Appin_projects

    def test_unique_values(self):
        df = pd.read_csv(fixture('iris_classification.csv'), na_values=['None'])
        unique_drgs = len(df.DRG.unique())
        test_df = impact_coding_on_a_single_column(df, 'species', 'DRG')
        unique_impact_values = len(test_df.DRG_impact_coded.unique())

        self.assertLessEqual(unique_impact_values, unique_drgs)

Example #2

    def test_column_renaming(self):
        df = pd.read_csv(fixture('iris_classification.csv'),
                         na_values=['None'])
        code_column_name = 'DRG'
        test_df = impact_coding_on_a_single_column(df, 'species',
                                                   code_column_name)

        self.assertTrue((code_column_name +
                         '_impact_coded') in test_df.columns)