Code Duplication    Length = 12-12 lines in 2 locations

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

@@ 2469-2480 (lines=12) @@
2466
        generator.
2467
    """
2468
2469
    def voltage_levels(p: float) -> int:
2470
        if p <= 100:
2471
            return 7
2472
        elif p <= 200:
2473
            return 6
2474
        elif p <= 5500:
2475
            return 5
2476
        elif p <= 20000:
2477
            return 4
2478
        elif p <= 120000:
2479
            return 3
2480
        return 1
2481
2482
    # Join mastr table
2483
    with db.session_scope() as session:

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

@@ 192-203 (lines=12) @@
189
            GeoDataFrame containing units all having assigned a voltage level.
190
        """
191
192
        def voltage_levels(p: float) -> int:
193
            if p <= 100:
194
                return 7
195
            elif p <= 200:
196
                return 6
197
            elif p <= 5500:
198
                return 5
199
            elif p <= 20000:
200
                return 4
201
            elif p <= 120000:
202
                return 3
203
            return 1
204
205
        units_gdf["voltage_level_inferred"] = False
206
        mask = units_gdf.voltage_level.isna()