def test_get_operator_class_not_exists(self):
with pytest.raises(KeyError):
get_operator_class('NotExists')
def test_get_operator_class(self, operator_class, return_value):
get_operator_class(operator_class) == return_value
