From 9644deaf6717aec5bc32343041c36c32a15d016d Mon Sep 17 00:00:00 2001
From: Daniel Olsen <daniel@breakthroughenergy.org>
Date: Mon, 4 Apr 2022 12:50:13 -0700
Subject: [PATCH] test: add test for add_b2bs_to_dc_lines

---
 .../data_process/tests/test_transmission.py     | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/prereise/gather/griddata/hifld/data_process/tests/test_transmission.py b/prereise/gather/griddata/hifld/data_process/tests/test_transmission.py
index 5a8c1f815..ca9459e9a 100644
--- a/prereise/gather/griddata/hifld/data_process/tests/test_transmission.py
+++ b/prereise/gather/griddata/hifld/data_process/tests/test_transmission.py
@@ -6,6 +6,7 @@
 
 from prereise.gather.griddata.hifld import const
 from prereise.gather.griddata.hifld.data_process.transmission import (
+    add_b2bs_to_dc_lines,
     add_impedance_and_rating,
     add_substation_info_to_buses,
     assign_buses_to_lines,
@@ -19,6 +20,22 @@
 )
 
 
+def test_add_b2bs_to_dc_lines():
+    dc_lines = pd.DataFrame(
+        {"SUB_1_ID": [1, 2], "SUB_2_ID": [3, 4], "Pmax": [5, 6]},
+        index=[200, 201],
+    )
+    substations = pd.DataFrame(
+        {"NAME": ["Wango_West", "Jango", "Wango_East", "Tango_South", "Tango_North"]}
+    )
+    b2b_ratings = {"Wango": 100, "Tango": 200}
+    expected_new_rows = pd.DataFrame(
+        {"SUB_1_ID": [0, 3], "SUB_2_ID": [2, 4], "Pmax": [100, 200]}, index=[300, 301]
+    )
+    add_b2bs_to_dc_lines(dc_lines, substations, b2b_ratings)
+    assert_frame_equal(dc_lines.iloc[2:], expected_new_rows)
+
+
 def test_add_impedance_and_rating():
     branch = pd.DataFrame(
         {