Add test (#346)

iiasa · Jul 7, 2020 · 51fbaaa · 51fbaaa
1 parent 9785f37
commit 51fbaaa
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ixmp/tests/core/test_timeseries.py b/ixmp/tests/core/test_timeseries.py
@@ -206,6 +206,13 @@ def test_add_timeseries(ts, fmt):
         ts.add_timeseries(DATA[0].drop('unit', axis=1))
 
 
+@pytest.mark.parametrize('fmt', ['long', 'wide'])
+def test_add_timeseries_unit(ts, fmt):
+    data = DATA[0].copy() if fmt == 'long' else wide(DATA[0])
+    data['unit'] = 1
+    ts.add_timeseries(data)
+
+
 @pytest.mark.parametrize('fmt', ['long', 'wide'])
 def test_add_and_remove_timeseries_with_long_variable_name(ts, fmt):
     data = (DATA[0] if fmt == 'long' else wide(DATA[0])).copy()