// SPDX-License-Identifier: MIT
// Copyright (c) "2023" . The DeepCausality Authors. All Rights Reserved.

use std::collections::BTreeMap;

use deep_causality::prelude::Observation;
use deep_causality::protocols::observable::ObservableReasoning;
use deep_causality::types::alias_types::NumericalValue;

use crate::utils::test_utils;

fn get_test_obs_btree_map() -> BTreeMap<usize, Observation> {
    let o1 = Observation::new(0, 10.0, 1.0);
    let o2 = Observation::new(1, 10.0, 1.0);
    let o3 = Observation::new(2, 10.0, 1.0);
    let o4 = Observation::new(3, 12.0, 0.0);
    let o5 = Observation::new(4, 14.0, 0.0);
    BTreeMap::from_iter([(1, o1), (2, o2), (3, o3), (4, o4), (5, o5)])
}

#[test]
fn test_add() {
    let mut col = get_test_obs_btree_map();
    assert_eq!(5, col.len());

    let o6 = test_utils::get_test_observation();
    col.insert(6, o6);
    assert_eq!(6, col.len());
}

#[test]
fn test_number_observation() {
    let observations = get_test_obs_btree_map();
    let target_threshold = 10.0 as NumericalValue;
    let target_effect = 1.0 as NumericalValue;
    let total_observation = observations.number_observation(target_threshold, target_effect);
    assert_eq!(3.0, total_observation);
}

#[test]
fn test_percent_observation() {
    let observations = get_test_obs_btree_map();
    let target_threshold = 10.0;
    let target_effect = 1.0;
    let percent_observation = observations.percent_observation(target_threshold, target_effect);
    assert_eq!(0.6, percent_observation);
}

#[test]
fn test_number_non_observation() {
    let observations = get_test_obs_btree_map();
    let target_threshold = 10.0 as NumericalValue;
    let target_effect = 1.0 as NumericalValue;
    let total_non_observation =
        observations.number_non_observation(target_threshold, target_effect);
    assert_eq!(2.0, total_non_observation);
}

#[test]
fn test_percent_non_observation() {
    let observations = get_test_obs_btree_map();
    let target_threshold = 10.0;
    let target_effect = 1.0;
    let percent_non_observation =
        observations.percent_non_observation(target_threshold, target_effect);
    assert_eq!(0.4, percent_non_observation);
}

#[test]
fn test_get_all_items() {
    let observations = get_test_obs_btree_map();
    let all_items = observations.get_all_items();

    let exp_len = observations.len();
    let act_len = all_items.len();
    assert_eq!(exp_len, act_len);
}

#[test]
fn test_len() {
    let col = get_test_obs_btree_map();
    assert_eq!(5, col.len());
}

#[test]
fn test_is_empty() {
    let col = get_test_obs_btree_map();
    assert!(!col.is_empty());
}