Elizaveta Lutsenko · @LizLutsenko · Rosneft rosneft · 4ede5787 · 45c37fa6 · 45c37fa6
--- a/test/test_node_cache.py
+++ b/test/test_node_cache.py
+   predictors, response = load_breast_cancer(return_X_y=True)
+   np.random.seed(1)
+   np.random.shuffle(predictors)
+   np.random.shuffle(response)
+   response = response[:100]
+   predictors = predictors[:100]
+   train_data_x, test_data_x = split_train_test(predictors)
+   train_data_y, test_data_y = split_train_test(response)
+   train_data = InputData(features=train_data_x, target=train_data_y,
+                          idx=np.arange(0, len(train_data_y)))
+   test_data = InputData(features=test_data_x, target=test_data_y,
+                         idx=np.arange(0, len(test_data_y)))
+   return train_data, test_data
+
+
+def chain_first():
--- a/test/test_node_cache.py
+++ b/test/test_node_cache.py
+   chain.fit(input_data=train)
+   other_chain.fit(input_data=train)
+   chain.replace_node(chain.root_node.nodes_from[0].nodes_from[0], other_chain.root_node.nodes_from[0])
+   root = chain.root_node
+   root_parent_first = root.nodes_from[0]
+   root_parent_second = root.nodes_from[1]
+   root_parent_first_parent_first = root_parent_first.nodes_from[0]
+   root_parent_first_parent_second = root_parent_first.nodes_from[1]
+   primary_node_first = root_parent_first_parent_first.nodes_from[0]
+   primary_node_second = root_parent_first_parent_first.nodes_from[0]
+
+   assert not all([root._is_cache_actual(), root_parent_first._is_cache_actual()])
+   assert all([root_parent_second._is_cache_actual(), root_parent_first_parent_first._is_cache_actual(),
+               root_parent_first_parent_second._is_cache_actual()])
+   assert all([primary_node_first._is_cache_actual(), primary_node_second._is_cache_actual()])
+
--- a/test/test_node_cache.py
+++ b/test/test_node_cache.py
+import numpy as np
+import pytest
+from core.models.data import InputData, split_train_test
+from core.composer.chain import Chain
+from core.composer.node import NodeGenerator
+from core.repository.model_types_repository import ModelTypesIdsEnum
+from core.repository.quality_metrics_repository import MetricsRepository, ClassificationMetricsEnum
+from sklearn.datasets import load_breast_cancer
+
+
+@pytest.fixture()
+def data_setup():