Skip to content

Commit 7cd2d44

Browse files
committed
Fix tests for broadcast_techs
1 parent 5eedbc0 commit 7cd2d44

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

tests/test_utilities.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,11 @@ def test_reduce_assets_with_zero_size(capacity: xr.DataArray):
4343
def test_broadcast_tech(technologies, capacity):
4444
from muse.utilities import broadcast_techs
4545

46+
technologies = technologies.sel(year=2010)
4647
regions = make_array(technologies.region)
4748
commodities = make_array(technologies.commodity)
48-
years = make_array(technologies.year)
4949
techs = make_array(technologies.technology)
50-
technologies["fixed_outputs"] = regions * commodities * years * techs
50+
technologies["fixed_outputs"] = regions * commodities * techs
5151

5252
actual = broadcast_techs(technologies.fixed_outputs, capacity)
5353

@@ -57,16 +57,15 @@ def test_broadcast_tech(technologies, capacity):
5757

5858
for asset in capacity.asset:
5959
region = regions.sel(region=asset.region)
60-
year = years.interp(year=asset.installed, method="linear")
6160
tech = techs.sel(technology=asset.technology)
62-
expected = region * year * tech * commodities
61+
expected = region * tech * commodities
6362
assert actual.isel(asset=int(asset)).values == approx(expected.values)
6463

6564

6665
def test_broadcast_tech_idempotent(technologies, capacity):
6766
from muse.utilities import broadcast_techs
6867

69-
first = broadcast_techs(technologies, capacity)
68+
first = broadcast_techs(technologies.sel(year=2010), capacity)
7069
second = broadcast_techs(first, capacity)
7170
assert (first == second).all()
7271

0 commit comments

Comments
 (0)