iframe-proxy | Sunbelt Computer Software

History

99 lines (79 loc) · 4.03 KB

Raw

import pickle

from unittest import mock

import numpy as np

import pytest

import ratapi.events

def test_event_register() -> None:

first_callback = mock.Mock()

second_callback = mock.Mock()

with pytest.raises(ValueError):

ratapi.events.register("Message", first_callback)

ratapi.events.register(ratapi.events.EventTypes.Message, first_callback)

result = ratapi.events.get_event_callback(ratapi.events.EventTypes.Message)

assert result == [first_callback]

ratapi.events.register(ratapi.events.EventTypes.Plot, second_callback)

assert ratapi.events.get_event_callback(ratapi.events.EventTypes.Plot) == [second_callback]

ratapi.events.register(ratapi.events.EventTypes.Message, second_callback)

# the list is not guaranteed to be in the same order as inputted hence the set

assert set(ratapi.events.get_event_callback(ratapi.events.EventTypes.Message)) == {first_callback, second_callback}

ratapi.events.clear(ratapi.events.EventTypes.Message, second_callback)

result = ratapi.events.get_event_callback(ratapi.events.EventTypes.Message)

assert result == [first_callback]

result = ratapi.events.get_event_callback(ratapi.events.EventTypes.Plot)

assert result == [second_callback]

ratapi.events.clear()

assert ratapi.events.get_event_callback(ratapi.events.EventTypes.Plot) == []

assert ratapi.events.get_event_callback(ratapi.events.EventTypes.Message) == []

def test_event_notify() -> None:

first_callback = mock.Mock()

second_callback = mock.Mock()

third_callback = mock.Mock()

ratapi.events.register(ratapi.events.EventTypes.Message, first_callback)

ratapi.events.register(ratapi.events.EventTypes.Plot, second_callback)

ratapi.events.register(ratapi.events.EventTypes.Progress, third_callback)

ratapi.events.notify(ratapi.events.EventTypes.Message, "Hello World")

first_callback.assert_called_once_with("Hello World")

second_callback.assert_not_called()

data = ratapi.events.PlotEventData()

ratapi.events.notify(ratapi.events.EventTypes.Plot, data)

first_callback.assert_called_once()

second_callback.assert_called_once_with(data)

data = ratapi.events.ProgressEventData()

ratapi.events.notify(ratapi.events.EventTypes.Progress, data)

first_callback.assert_called_once()

second_callback.assert_called_once()

third_callback.assert_called_once_with(data)

ratapi.events.clear()

ratapi.events.notify(ratapi.events.EventTypes.Message, "Hello World")

ratapi.events.notify(ratapi.events.EventTypes.Plot, data)

assert first_callback.call_count == 1

assert second_callback.call_count == 1

assert third_callback.call_count == 1

def test_event_data_pickle():

data = ratapi.events.ProgressEventData()

data.message = "Hello"

data.percent = 0.5

pickled_data = pickle.loads(pickle.dumps(data))

assert pickled_data.message == data.message

assert pickled_data.percent == data.percent

data = ratapi.events.PlotEventData()

data.modelType = "custom layers"

data.dataPresent = np.ones(2)

data.subRoughs = np.ones((20, 2))

data.resample = np.ones(2)

data.resampledLayers = [np.ones((20, 2)), np.ones((20, 2))]

data.reflectivity = [np.ones((20, 2)), np.ones((20, 2))]

data.shiftedData = [np.ones((20, 2)), np.ones((20, 2))]

data.sldProfiles = [np.ones((20, 2)), np.ones((20, 2))]

data.contrastNames = ["D2O", "SMW"]

pickled_data = pickle.loads(pickle.dumps(data))

assert pickled_data.modelType == data.modelType

assert (pickled_data.dataPresent == data.dataPresent).all()

assert (pickled_data.subRoughs == data.subRoughs).all()

assert (pickled_data.resample == data.resample).all()

for i in range(2):

assert (pickled_data.resampledLayers[i] == data.resampledLayers[i]).all()

assert (pickled_data.reflectivity[i] == data.reflectivity[i]).all()

assert (pickled_data.shiftedData[i] == data.shiftedData[i]).all()

assert (pickled_data.sldProfiles[i] == data.sldProfiles[i]).all()

assert pickled_data.contrastNames == data.contrastNames

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

test_events.py

Latest commit

History

test_events.py

File metadata and controls