File size: 1,959 Bytes
8b37c3f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | from __future__ import annotations
import numpy as np
import pandas as pd
from src.data.partial_cycle import (
PartialCycleConfig,
add_soh_labels,
extract_partial_cycle_features,
make_partial_cycle_sequence,
prior_equivalent_full_cycles,
select_partial_discharge,
)
from src.utils.config import FEATURE_COLS_V3, SEQUENCE_FEATURE_COLS_V3
def _cycle() -> pd.DataFrame:
return pd.DataFrame({
"Time": np.arange(8, dtype=float) * 10.0,
"Voltage_measured": [4.2, 4.05, 3.99, 3.90, 3.80, 3.70, 3.59, 3.40],
"Current_measured": [-2.0] * 8,
"Temperature_measured": np.linspace(25.0, 27.0, 8),
})
def test_soh_reference_is_median_of_first_three_valid_cycles():
cycles = pd.DataFrame({
"battery_id": ["B1"] * 4,
"cycle_number": [0, 1, 2, 3],
"Capacity": [2.0, 1.98, 1.96, 1.80],
})
labelled = add_soh_labels(cycles)
assert np.isclose(labelled["reference_capacity_ah"].iloc[0], 1.98)
assert np.isclose(labelled["SoH"].iloc[-1], 100 * 1.80 / 1.98)
def test_prior_throughput_excludes_current_cycle():
efc = prior_equivalent_full_cycles(pd.Series([2.0, 1.8, 1.6]), 2.0)
assert np.allclose(efc, [0.0, 1.0, 1.9])
def test_partial_feature_and_sequence_contracts():
segment = select_partial_discharge(_cycle())
assert segment["Voltage_measured"].iloc[0] <= 4.0
assert segment["Voltage_measured"].iloc[-1] <= 3.6
features = extract_partial_cycle_features(
_cycle(), cycle_index=7, prior_efc=6.2, ambient_temperature=24.0,
rated_capacity_ah=2.0,
)
assert list(features) == FEATURE_COLS_V3
assert "Capacity" not in features and "SoH" not in features
sequence = make_partial_cycle_sequence(
_cycle(), ambient_temperature=24.0, rated_capacity_ah=2.0,
config=PartialCycleConfig(sequence_bins=64),
)
assert sequence.shape == (64, 5)
assert list(sequence) == SEQUENCE_FEATURE_COLS_V3
|