Skip to content

Commit

Permalink
test: remove tests for old impedance/ratings methods
Browse files Browse the repository at this point in the history
  • Loading branch information
danielolsen committed Apr 5, 2022
1 parent 77ce876 commit deb3357
Showing 1 changed file with 0 additions and 65 deletions.
Original file line number Diff line number Diff line change
@@ -1,10 +1,7 @@
from unittest.mock import Mock

import pandas as pd
import pytest
from pandas.testing import assert_frame_equal, assert_series_equal

from prereise.gather.griddata.hifld import const
from prereise.gather.griddata.hifld.data_process.transmission import (
add_b2bs_to_dc_lines,
add_lines_impedances_ratings,
Expand All @@ -13,8 +10,6 @@
augment_line_voltages,
create_buses,
create_transformers,
estimate_branch_impedance,
estimate_branch_rating,
filter_islands_and_connect_with_mst,
map_lines_to_substations_using_coords,
split_lines_to_ac_and_dc,
Expand Down Expand Up @@ -222,66 +217,6 @@ def test_create_transformers():
assert_frame_equal(transformers, expected_transformers)


def test_estimate_branch_impedance_lines():
resistance = 0.01
reactance = 0.1
fake_lines = [Mock(series_impedance=(resistance + 1j * reactance))] * 3
branch = pd.DataFrame(
{"VOLTAGE": [69, 70, 345], "type": ["Line"] * 3, "line_object": fake_lines}
)
x = estimate_branch_impedance(branch.iloc[0], pd.Series())
assert x == reactance / (69**2 / const.s_base)
x = estimate_branch_impedance(branch.iloc[1], pd.Series())
assert x == reactance / (70**2 / const.s_base)
x = estimate_branch_impedance(branch.iloc[2], pd.Series())
assert x == reactance / (345**2 / const.s_base)


def test_estimate_branch_impedance_transformers():
transformers = pd.DataFrame(
{"from_bus_id": [0, 1, 2], "to_bus_id": [1, 2, 3], "type": ["Transformer"] * 3}
)
bus_voltages = pd.Series([69, 230, 350, 500])
x = estimate_branch_impedance(transformers.iloc[0], bus_voltages)
assert x == const.transformer_reactance[(69, 230)]
x = estimate_branch_impedance(transformers.iloc[1], bus_voltages)
assert x == const.transformer_reactance[(230, 345)]
x = estimate_branch_impedance(transformers.iloc[2], bus_voltages)
assert x == const.transformer_reactance[(345, 500)]


def test_estimate_branch_rating_lines():
fake_ratings = pd.Series([10, 20, 30, 40])
fake_thermal_ratings = pd.Series([100, 200, 300, 400])
fake_lines = [Mock(power_rating=i) for i in fake_ratings]
branch = pd.DataFrame(
{
"VOLTAGE": [69, 140, 345, 499],
"type": ["Line"] * 4,
"line_object": fake_lines,
}
)
assert_series_equal(
fake_ratings,
branch.apply(estimate_branch_rating, args=[None, fake_thermal_ratings], axis=1),
)


def test_estimate_branch_rating_transformers():
thermal_ratings = pd.Series([100, 550, 1655, 2585], index=[69, 230, 345, 500])
transformers = pd.DataFrame(
{"from_bus_id": [0, 1, 2], "to_bus_id": [1, 2, 3], "type": ["Transformer"] * 3}
)
bus_voltages = pd.Series([69, 230, 350, 500])

rating = estimate_branch_rating(transformers.iloc[0], bus_voltages, thermal_ratings)
assert rating == const.transformer_rating
rating = estimate_branch_rating(transformers.iloc[1], bus_voltages, thermal_ratings)
assert rating == const.transformer_rating * 3
rating = estimate_branch_rating(transformers.iloc[2], bus_voltages, thermal_ratings)
assert rating == const.transformer_rating * 4


def test_filter_islands_and_connect_with_mst():
lines = pd.DataFrame(
{
Expand Down

0 comments on commit deb3357

Please sign in to comment.