def test_pipeline_validate_entities(t, l, i, b, se):
p = Pipeline()
data_type = [t, l, i, b, se]
for data in data_type:
with pytest.raises(TypeError):
p._validate_entities(data)
s = Stage()
assert isinstance(p._validate_entities(s), list)
s1 = Stage()
s2 = Stage()
assert [s1,s2] == p._validate_entities([s1,s2])
def test_pipeline_validate_entities(t, l, i, b, se):
p = Pipeline()
data_type = [t, l, i, b, se]
for data in data_type:
with pytest.raises(TypeError):
p._validate_entities(data)
s = Stage()
assert isinstance(p._validate_entities(s), list)
s1 = Stage()
s2 = Stage()
assert [s1,s2] == p._validate_entities([s1,s2])
def test_pipeline_validate_entities(self, mocked_init, t, l, i, b, se):
p = Pipeline()
data_type = [t, l, i, b, se]
for data in data_type:
with self.assertRaises(TypeError):
p._validate_entities(data)
s = mock.MagicMock(spec=Stage)
self.assertIsInstance(p._validate_entities(s), list)
s1 = mock.MagicMock(spec=Stage)
s2 = mock.MagicMock(spec=Stage)
self.assertEqual([s1,s2], p._validate_entities([s1,s2]))
