feat: add substation transmission capacity calculation

[danielolsen]

Pull Request doc

Purpose

Add a feature to calculate the maximum transmission capacity into or out of a substation. This can be helpful e.g. when working to troubleshoot feasibility for a new grid model, when designing a Grid for a new scenario, or when investigating capacity expansion candidates.

What the code is doing

In substations.py:

• We create a copy of the grid.branch data frame which has additional columns for from_sub and to_sub.
• We group on these from_sub and to_sub columns, and sum the branch capacities for branches 'from' and 'to' each substation, excluding branches which are 'from' and 'to' the same substation.
• We reindex so that all substation IDs are present in each of the 'from' and 'to' summations.
• Now that the indices are aligned, we sum the 'from' and 'to' summations and return.

Testing

New unit test added. It features branches which have opposite to/from bus IDs (summed appropriately), and one branch with to/from buses within the same substation (not counted in any substation's capacity).

Usage Example/Visuals

>>> from powersimdata.design.transmission.substations import calculate_substation_capacity
>>> from powersimdata import Grid
>>> grid = Grid("Western")
Reading bus.csv
Reading plant.csv
Reading gencost.csv
Reading branch.csv
Reading dcline.csv
Reading sub.csv
Reading bus2sub.csv
Reading zone.csv
>>> calculate_substation_capacity(grid)
sub_id
35000    453.21
35001    492.76
35002    584.01
35003    465.77
35004    662.52
          ...
41078    100.00
41079    100.00
41080    100.00
41081    100.00
41082    100.00
Length: 4786, dtype: float64

Time estimate

15 minutes.

[BainanXia]

Looks good and thanks for adding tests!

[ywang303]

looks great! Do we have zone level calculate_capacity built-in method?

[danielolsen]

looks great! Do we have zone level calculate_capacity built-in method?

Not yet, no. I believe it would be relatively straightforward to implement in the same way: we would just replace the substation lookup with a zone lookup.

[ywang303]

cool! Get Outlook for iOS<https://aka.ms/o0ukef>

…

________________________________ From: danielolsen ***@***.***> Sent: Tuesday, June 15, 2021 4:59:08 PM To: Breakthrough-Energy/PowerSimData ***@***.***> Cc: Josephine Wang ***@***.***>; Review requested ***@***.***> Subject: Re: [Breakthrough-Energy/PowerSimData] feat: add substation transmission capacity calculation (#500) looks great! Do we have zone level calculate_capacity built-in method? Not yet, no. I believe it would be relatively straightforward to implement in the same way: we would just replace the substation lookup with a zone lookup. — You are receiving this because your review was requested. Reply to this email directly, view it on GitHub<https://nam02.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.jparrowsec.cn%2FBreakthrough-Energy%2FPowerSimData%2Fpull%2F500%23issuecomment-861826827&data=04%7C01%7Cywang303%40jhu.edu%7Cacac3a2eaf14400f508208d930406bee%7C9fa4f438b1e6473b803f86f8aedf0dec%7C0%7C0%7C637593875513350502%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C1000&sdata=eo0rC%2BdymKHmIJLkA3hKUzxUXcOYfKusJc8xOZWC%2Fe4%3D&reserved=0>, or unsubscribe<https://nam02.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.jparrowsec.cn%2Fnotifications%2Funsubscribe-auth%2FAOQTGHJUA26IIK2K7TD5D23TS65JZANCNFSM46WFDVOQ&data=04%7C01%7Cywang303%40jhu.edu%7Cacac3a2eaf14400f508208d930406bee%7C9fa4f438b1e6473b803f86f8aedf0dec%7C0%7C0%7C637593875513350502%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C1000&sdata=7Vm8aL5ZNn97XFEH04wtF%2FJX0omtxXg5SEIzXoxve%2Bg%3D&reserved=0>.