def test__get_intervals(self): # Run data = pd.Series(['foo', 'bar', 'bar', 'foo', 'foo', 'tar']) result = CategoricalTransformer._get_intervals(data) # Asserts expected_intervals = { 'foo': (0, 0.5, 0.25, 0.5 / 6), 'bar': (0.5, 0.8333333333333333, 0.6666666666666666, 0.05555555555555555), 'tar': (0.8333333333333333, 0.9999999999999999, 0.9166666666666666, 0.027777777777777776) } assert result[0] == expected_intervals
def test__get_intervals(self): """Test get category intervals""" # Setup data = pd.Series(['bar', 'foo', 'foo', 'tar']) # Run result = CategoricalTransformer._get_intervals(data) # Asserts expected_intervals = { 'foo': (0, 0.5), 'tar': (0.5, 0.75), 'bar': (0.75, 1) } assert result == expected_intervals