Skip to content
Snippets Groups Projects
Commit 42bd3e20 authored by Oscar Gustafsson's avatar Oscar Gustafsson :bicyclist:
Browse files

Add first test for operation split

parent eae8191f
No related branches found
No related tags found
No related merge requests found
Pipeline #88148 passed
...@@ -735,7 +735,7 @@ class AbstractOperation(Operation, AbstractGraphComponent): ...@@ -735,7 +735,7 @@ class AbstractOperation(Operation, AbstractGraphComponent):
try: try:
result = self.evaluate(*([Input()] * self.input_count)) result = self.evaluate(*([Input()] * self.input_count))
if isinstance(result, collections.Sequence) and all( if isinstance(result, collections.abc.Sequence) and all(
isinstance(e, Operation) for e in result isinstance(e, Operation) for e in result
): ):
return result return result
......
...@@ -310,3 +310,12 @@ class TestIOCoordinates: ...@@ -310,3 +310,12 @@ class TestIOCoordinates:
i_c, o_c = bfly.get_io_coordinates() i_c, o_c = bfly.get_io_coordinates()
assert i_c == [[2, 0.25], [3, 0.75]] assert i_c == [[2, 0.25], [3, 0.75]]
assert o_c == [[5, 0.25], [10, 0.75]] assert o_c == [[5, 0.25], [10, 0.75]]
class TestSplit:
def test_simple_case(self):
bfly = Butterfly()
split = bfly.split()
assert len(split) == 2
assert sum(isinstance(op, Addition) for op in split) == 1
assert sum(isinstance(op, Subtraction) for op in split) == 1
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment