32 lines
772 B
Python
32 lines
772 B
Python
import pdme.inputs
|
|
|
|
|
|
def test_inputs_with_frequency_range():
|
|
dot1 = [1, 2, 3]
|
|
dot2 = [2, 4, 6]
|
|
|
|
frequencies = [5, 7, 9]
|
|
|
|
expected = [
|
|
([1, 2, 3], 5), ([1, 2, 3], 7), ([1, 2, 3], 9), ([2, 4, 6], 7), ([2, 4, 6], 9), ([2, 4, 6], 5)
|
|
]
|
|
|
|
actual = pdme.inputs.inputs_with_frequency_range([dot1, dot2], frequencies)
|
|
|
|
assert sorted(actual) == sorted(expected), "Did not actually match dot inputs!"
|
|
|
|
|
|
def test_input_pairs_with_frequency_range():
|
|
dot1 = [1, 2, 3]
|
|
dot2 = [2, 4, 6]
|
|
|
|
frequencies = [5, 7, 9]
|
|
|
|
expected = [
|
|
([1, 2, 3], [2, 4, 6], 5), ([1, 2, 3], [2, 4, 6], 7), ([1, 2, 3], [2, 4, 6], 9)
|
|
]
|
|
|
|
actual = pdme.inputs.input_pairs_with_frequency_range([dot1, dot2], frequencies)
|
|
|
|
assert sorted(actual) == sorted(expected), "Did not actually match dot inputs!"
|