Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
dist=Distribution()
)
another_primitives_ep = EntryPoint(
'primitives',
'tests.test_discovery',
attrs=['FAKE_PRIMITIVES_PATHS'],
dist=Distribution()
)
iep_mock.return_value = [
something_else_ep,
primitives_ep,
another_primitives_ep
]
# run
paths = discovery._load_entry_points('primitives')
# assert
expected = [
'this/is/a/fake',
'this/is/another/fake',
'this/is/yet/another/fake',
]
assert paths == expected
expected_calls = [
call('mlblocks'),
]
assert iep_mock.call_args_list == expected_calls
def test__load_entry_points_no_entry_points(iep_mock):
# setup
iep_mock.return_value == []
# run
paths = discovery._load_entry_points('jsons_path', 'mlprimitives')
# assert
assert paths == []
expected_calls = [
call('mlprimitives'),
]
assert iep_mock.call_args_list == expected_calls