"""Tests for logic utils.""" import pytest from salessheets.utils import logic_utils @pytest.mark.parametrize( 'user_id, expected_result', [ ('oa:23487', False), ('alw:76543', True), (None, False)]) def test_is_workstation_user(user_id, expected_result): """Expect function given the user_id return correct boolean.""" result = logic_utils.is_workstation_user(user_id) assert result == expected_result