def test_evaluate():
data = np.random.rand(60).reshape((15, 4))
labels = np.random.randint(0, 2, 15)
data_node = DataNode()
data_node._model = Mock()
data_node.evaluate(data, labels)
data_node._model.evaluate.assert_called_once_with(data, labels)
def test_performance():
data_node = DataNode()
data_node._model = Mock()
data_node._model.performance.return_value = 0
data = np.random.rand(25).reshape((5, 5))
labels = np.random.randint(0, 2, 5)
res = data_node.performance(data, labels)
data_node._model.performance.assert_called_once_with(data, labels)
assert res == 0
def test_local_evaluate():
data_key = 'EMNIST'
data_node = DataNode()
data_node._private_test_data = Mock()
data = Mock()
data.data = np.random.rand(60)
data.label = np.random.randint(0, 2, 60)
data_node._private_test_data.get.return_value = data
data_node._model = Mock()
data_node._model.evaluate.return_value = 0
data_node.self_private_test_data = 1
eval = data_node.local_evaluate(data_key)
assert eval == 0
data_node._private_test_data.get.assert_called_once_with(data_key)
data_node._model.evaluate.assert_called_once_with(data.data, data.label)