Code Duplication    Length = 12-12 lines in 2 locations

src/egon/data/datasets/power_plants/mastr.py 1 location

@@ 130-141 (lines=12) @@
127
        GeoDataFrame containing units all having assigned a voltage level.
128
    """
129
130
    def voltage_levels(p: float) -> int:
131
        if p <= 100:
132
            return 7
133
        elif p <= 200:
134
            return 6
135
        elif p <= 5500:
136
            return 5
137
        elif p <= 20000:
138
            return 4
139
        elif p <= 120000:
140
            return 3
141
        return 1
142
143
    units_gdf["voltage_level_inferred"] = False
144
    mask = units_gdf.voltage_level.isna()

src/egon/data/datasets/power_plants/pv_rooftop_buildings.py 1 location

@@ 1971-1982 (lines=12) @@
1968
        generator.
1969
    """
1970
1971
    def voltage_levels(p: float) -> int:
1972
        if p <= 100:
1973
            return 7
1974
        elif p <= 200:
1975
            return 6
1976
        elif p <= 5500:
1977
            return 5
1978
        elif p <= 20000:
1979
            return 4
1980
        elif p <= 120000:
1981
            return 3
1982
        return 1
1983
1984
    # Join mastr table
1985
    with db.session_scope() as session: