"""
# -*- coding: utf-8 -*-
#
# Copyright 2021 Michael Büsch <m@bues.ch>
#
# Licensed under the Apache License version 2.0
# or the MIT license, at your option.
# SPDX-License-Identifier: Apache-2.0 OR MIT
#
"""

from mlplib.batch import *
import numpy as np

def test_minibatches():
    x = np.array([[1, 2],
                  [3, 4],
                  [5, 6],
                  [7, 8],
                  [9, 10]])
    y = np.array([[10, 20],
                  [30, 40],
                  [50, 60],
                  [70, 80],
                  [90, 100]])
    m = list(minibatches(x, y, 2))
    assert np.all(m[0][0] == np.array([[1, 2], [3, 4]]))
    assert np.all(m[1][0] == np.array([[5, 6], [7, 8]]))
    assert np.all(m[2][0] == np.array([[9, 10]]))
    assert np.all(m[0][1] == np.array([[10, 20], [30, 40]]))
    assert np.all(m[1][1] == np.array([[50, 60], [70, 80]]))
    assert np.all(m[2][1] == np.array([[90, 100]]))

def test_proportion_equal():
    a = np.array([[1, 2, 3], [4, 5, 6]])
    b = np.array([[1.1, 2, 3.3], [4, -5, 6]])
    p = proportion_equal(a, b)
    assert p == 0.5

def test_array_to_boolint():
    a = array_to_boolint(np.array([[0.0, 0.1, -0.1, 0.74, 0.75, 0.76, 0.9, 1.0],
                                   [0.9, 0.5, 0.7, 0.6, 0.5, 0.4, 0.8, 0.1]]))
    assert np.all(a == np.array([[0, 0, 0, 0, 1, 1, 1, 1],
                                 [1, 0, 0, 0, 0, 0, 1, 0]]))

def test_collapse_bool_nodes():
    a = collapse_bool_nodes(np.array([[0.9, 0.1, 0.1],
                                      [0.1, 0.9, 0.1],
                                      [0.1, 0.1, 0.9],
                                      [0.9, 0.9, 0.9]]))
    assert np.all(a == np.array([1, 2, 4, 7]))

# vim: ts=4 sw=4 expandtab