# Base Dependencies
# ----------------
import pytest
import numpy as np
from random import randrange

# Local Dependencies
# ------------------
from models import RelationCollection
from features.position_feature import PositionFeature

# Constants
# ---------
from constants import (
    N2C2_ATTR_ENTITY_CANDIDATES,
    DDI_ATTR_ENTITY_CANDIDATES,
)


# Tests
# ------
def test_position_init():
    position = PositionFeature("n2c2")
    assert isinstance(position, PositionFeature)
    assert position.dataset == "n2c2"
    assert position.attr_entity_candidates == N2C2_ATTR_ENTITY_CANDIDATES

    position = PositionFeature("ddi")
    assert isinstance(position, PositionFeature)
    assert position.dataset == "ddi"
    assert position.attr_entity_candidates == DDI_ATTR_ENTITY_CANDIDATES


def test_position_init_unsupported_dataset_raises():
    with pytest.raises(ValueError):
        position = PositionFeature("i2b2")


def test_position_get_feature_names():
    position = PositionFeature("n2c2")
    assert len(position.get_feature_names()) == 2


def test_position_fit(n2c2_small_collection):
    position = PositionFeature("n2c2")
    position = position.fit(n2c2_small_collection)
    assert isinstance(position, PositionFeature)


def test_position_create_position_feature_(n2c2_small_collection, relation):

    position = PositionFeature("n2c2")
    position_feature = position.create_position_feature(n2c2_small_collection)
    assert position_feature.shape == (len(n2c2_small_collection), 2)

    for i in range(len(n2c2_small_collection)):
        assert (position_feature[i][0] == 0 and position_feature[i][1] >= 0) or (
            position_feature[i][0] <= 0 and position_feature[i][1] == 0
        )
        
    one_collection = RelationCollection(relation)
    
    position_feature = position.create_position_feature(one_collection)
    
    assert (position_feature == np.array([[-1, 0]])).all()