How to use make method in lisa

Best Python code snippet using lisa_python

test_user.py

Source:test_user.py Github

copy

Full Screen

1from unittest.mock import Mock, patch2import pytest3from django.core.exceptions import ObjectDoesNotExist4from django.urls import reverse5from django.utils import timezone6from rest_framework import status7from rest_framework.response import Response8from rest_framework.test import APIClient9from apps.base.constants import ADMIN_PERMISSIONS, EDITOR_PERMISSIONS10from apps.base.models import UserNotificationPolicy11from apps.user_management.models.user import default_working_hours12from common.constants.role import Role13@pytest.mark.django_db14def test_update_user(15 make_organization,16 make_team,17 make_user_for_organization,18 make_token_for_organization,19 make_user_auth_headers,20):21 organization = make_organization()22 admin = make_user_for_organization(organization)23 team = make_team(organization)24 team.users.add(admin)25 _, token = make_token_for_organization(organization)26 client = APIClient()27 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})28 data = {29 "unverified_phone_number": "+79123456789",30 "current_team": team.public_primary_key,31 }32 response = client.put(url, data, format="json", **make_user_auth_headers(admin, token))33 assert response.status_code == status.HTTP_200_OK34 assert response.json()["unverified_phone_number"] == data["unverified_phone_number"]35 assert response.json()["current_team"] == data["current_team"]36@pytest.mark.django_db37def test_update_user_cant_change_email_and_username(38 make_organization,39 make_user_for_organization,40 make_token_for_organization,41 make_user_auth_headers,42):43 organization = make_organization()44 admin = make_user_for_organization(organization)45 _, token = make_token_for_organization(organization)46 client = APIClient()47 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})48 phone_number = "+79123456789"49 data = {50 "unverified_phone_number": phone_number,51 "email": "test@amixr.io",52 "username": "bob",53 }54 expected_response = {55 "pk": admin.public_primary_key,56 "organization": {"pk": organization.public_primary_key, "name": organization.org_title},57 "current_team": None,58 "email": admin.email,59 "hide_phone_number": False,60 "username": admin.username,61 "role": admin.role,62 "timezone": None,63 "working_hours": default_working_hours(),64 "unverified_phone_number": phone_number,65 "verified_phone_number": None,66 "telegram_configuration": None,67 "messaging_backends": {68 "TESTONLY": {69 "user": admin.username,70 }71 },72 "cloud_connection_status": 0,73 "permissions": ADMIN_PERMISSIONS,74 "notification_chain_verbal": {"default": "", "important": ""},75 "slack_user_identity": None,76 "avatar": admin.avatar_url,77 }78 response = client.put(url, data, format="json", **make_user_auth_headers(admin, token))79 assert response.status_code == status.HTTP_200_OK80 assert response.json() == expected_response81@pytest.mark.django_db82def test_list_users(83 make_organization,84 make_user_for_organization,85 make_token_for_organization,86 make_user_auth_headers,87):88 organization = make_organization()89 admin = make_user_for_organization(organization)90 editor = make_user_for_organization(organization, role=Role.EDITOR)91 _, token = make_token_for_organization(organization)92 client = APIClient()93 url = reverse("api-internal:user-list")94 expected_payload = {95 "count": 2,96 "next": None,97 "previous": None,98 "results": [99 {100 "pk": admin.public_primary_key,101 "organization": {"pk": organization.public_primary_key, "name": organization.org_title},102 "current_team": None,103 "email": admin.email,104 "hide_phone_number": False,105 "username": admin.username,106 "role": admin.role,107 "timezone": None,108 "working_hours": default_working_hours(),109 "unverified_phone_number": None,110 "verified_phone_number": None,111 "telegram_configuration": None,112 "messaging_backends": {113 "TESTONLY": {114 "user": admin.username,115 }116 },117 "permissions": ADMIN_PERMISSIONS,118 "notification_chain_verbal": {"default": "", "important": ""},119 "slack_user_identity": None,120 "avatar": admin.avatar_url,121 "cloud_connection_status": 0,122 },123 {124 "pk": editor.public_primary_key,125 "organization": {"pk": organization.public_primary_key, "name": organization.org_title},126 "current_team": None,127 "email": editor.email,128 "hide_phone_number": False,129 "username": editor.username,130 "role": editor.role,131 "timezone": None,132 "working_hours": default_working_hours(),133 "unverified_phone_number": None,134 "verified_phone_number": None,135 "telegram_configuration": None,136 "messaging_backends": {137 "TESTONLY": {138 "user": editor.username,139 }140 },141 "permissions": EDITOR_PERMISSIONS,142 "notification_chain_verbal": {"default": "", "important": ""},143 "slack_user_identity": None,144 "avatar": editor.avatar_url,145 "cloud_connection_status": 0,146 },147 ],148 }149 response = client.get(url, format="json", **make_user_auth_headers(admin, token))150 assert response.status_code == status.HTTP_200_OK151 assert response.json() == expected_payload152@pytest.mark.django_db153def test_notification_chain_verbal(154 make_organization,155 make_user_for_organization,156 make_token_for_organization,157 make_user_auth_headers,158 make_user_notification_policy,159):160 organization = make_organization()161 admin = make_user_for_organization(organization)162 _, token = make_token_for_organization(organization)163 data_for_creation = [164 {"step": UserNotificationPolicy.Step.NOTIFY, "notify_by": UserNotificationPolicy.NotificationChannel.SLACK},165 {"step": UserNotificationPolicy.Step.WAIT, "wait_delay": timezone.timedelta(minutes=5)},166 {167 "step": UserNotificationPolicy.Step.NOTIFY,168 "notify_by": UserNotificationPolicy.NotificationChannel.PHONE_CALL,169 },170 {"step": UserNotificationPolicy.Step.WAIT, "wait_delay": None},171 {"step": UserNotificationPolicy.Step.NOTIFY, "notify_by": UserNotificationPolicy.NotificationChannel.TELEGRAM},172 {"step": None},173 {174 "step": UserNotificationPolicy.Step.NOTIFY,175 "notify_by": UserNotificationPolicy.NotificationChannel.SLACK,176 "important": True,177 },178 {"step": UserNotificationPolicy.Step.WAIT, "wait_delay": timezone.timedelta(minutes=5), "important": True},179 {180 "step": UserNotificationPolicy.Step.NOTIFY,181 "notify_by": UserNotificationPolicy.NotificationChannel.PHONE_CALL,182 "important": True,183 },184 {"step": UserNotificationPolicy.Step.WAIT, "wait_delay": None, "important": True},185 {186 "step": UserNotificationPolicy.Step.NOTIFY,187 "notify_by": UserNotificationPolicy.NotificationChannel.TELEGRAM,188 "important": True,189 },190 ]191 for data in data_for_creation:192 make_user_notification_policy(admin, **data)193 client = APIClient()194 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})195 expected_notification_chain = {196 "default": "Slack - 5 min - \U0000260E - Telegram",197 "important": "Slack - 5 min - \U0000260E - Telegram",198 }199 response = client.get(url, format="json", **make_user_auth_headers(admin, token))200 assert response.status_code == status.HTTP_200_OK201 assert response.json()["notification_chain_verbal"] == expected_notification_chain202@pytest.mark.django_db203@pytest.mark.parametrize(204 "role,expected_status",205 [206 (Role.ADMIN, status.HTTP_200_OK),207 (Role.EDITOR, status.HTTP_200_OK),208 (Role.VIEWER, status.HTTP_403_FORBIDDEN),209 ],210)211def test_user_update_self_permissions(212 make_organization,213 make_user_for_organization,214 make_token_for_organization,215 make_user_auth_headers,216 role,217 expected_status,218):219 organization = make_organization()220 tester = make_user_for_organization(organization, role=role)221 _, token = make_token_for_organization(organization)222 client = APIClient()223 url = reverse("api-internal:user-detail", kwargs={"pk": tester.public_primary_key})224 with patch(225 "apps.api.views.user.UserView.update",226 return_value=Response(227 status=status.HTTP_200_OK,228 ),229 ):230 response = client.put(url, format="json", **make_user_auth_headers(tester, token))231 assert response.status_code == expected_status232@pytest.mark.django_db233@pytest.mark.parametrize(234 "role,expected_status",235 [236 (Role.ADMIN, status.HTTP_200_OK),237 (Role.EDITOR, status.HTTP_403_FORBIDDEN),238 (Role.VIEWER, status.HTTP_403_FORBIDDEN),239 ],240)241def test_user_update_other_permissions(242 make_organization,243 make_user_for_organization,244 make_token_for_organization,245 make_user_auth_headers,246 role,247 expected_status,248):249 organization = make_organization()250 admin = make_user_for_organization(organization)251 tester = make_user_for_organization(organization, role=role)252 _, token = make_token_for_organization(organization)253 client = APIClient()254 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})255 data = {"unverified_phone_number": "+79123456789"}256 response = client.put(url, data, format="json", **make_user_auth_headers(tester, token))257 assert response.status_code == expected_status258@pytest.mark.django_db259@pytest.mark.parametrize(260 "role,expected_status",261 [262 (Role.ADMIN, status.HTTP_200_OK),263 (Role.EDITOR, status.HTTP_200_OK),264 (Role.VIEWER, status.HTTP_403_FORBIDDEN),265 ],266)267def test_user_list_permissions(268 make_organization,269 make_user_for_organization,270 make_token_for_organization,271 make_user_auth_headers,272 role,273 expected_status,274):275 organization = make_organization()276 tester = make_user_for_organization(organization, role=role)277 _, token = make_token_for_organization(organization)278 client = APIClient()279 url = reverse("api-internal:user-list")280 with patch(281 "apps.api.views.user.UserView.list",282 return_value=Response(283 status=status.HTTP_200_OK,284 ),285 ):286 response = client.get(url, format="json", **make_user_auth_headers(tester, token))287 assert response.status_code == expected_status288@pytest.mark.django_db289@pytest.mark.parametrize(290 "role,expected_status",291 [292 (Role.ADMIN, status.HTTP_200_OK),293 (Role.EDITOR, status.HTTP_200_OK),294 (Role.VIEWER, status.HTTP_200_OK),295 ],296)297def test_user_detail_self_permissions(298 make_organization,299 make_user_for_organization,300 make_token_for_organization,301 make_user_auth_headers,302 role,303 expected_status,304):305 organization = make_organization()306 tester = make_user_for_organization(organization, role=role)307 _, token = make_token_for_organization(organization)308 client = APIClient()309 url = reverse("api-internal:user-detail", kwargs={"pk": tester.public_primary_key})310 with patch(311 "apps.api.views.user.UserView.list",312 return_value=Response(313 status=status.HTTP_200_OK,314 ),315 ):316 response = client.get(url, format="json", **make_user_auth_headers(tester, token))317 assert response.status_code == expected_status318@pytest.mark.django_db319@pytest.mark.parametrize(320 "role,expected_status",321 [322 (Role.ADMIN, status.HTTP_200_OK),323 (Role.EDITOR, status.HTTP_403_FORBIDDEN),324 (Role.VIEWER, status.HTTP_403_FORBIDDEN),325 ],326)327def test_user_detail_other_permissions(328 make_organization,329 make_user_for_organization,330 make_token_for_organization,331 make_user_auth_headers,332 role,333 expected_status,334):335 organization = make_organization()336 admin = make_user_for_organization(organization)337 tester = make_user_for_organization(organization, role=role)338 _, token = make_token_for_organization(organization)339 client = APIClient()340 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})341 response = client.get(url, format="json", **make_user_auth_headers(tester, token))342 assert response.status_code == expected_status343@pytest.mark.django_db344@pytest.mark.parametrize(345 "role,expected_status",346 [347 (Role.ADMIN, status.HTTP_200_OK),348 (Role.EDITOR, status.HTTP_200_OK),349 (Role.VIEWER, status.HTTP_403_FORBIDDEN),350 ],351)352def test_user_get_own_verification_code(353 make_organization,354 make_user_for_organization,355 make_token_for_organization,356 make_user_auth_headers,357 role,358 expected_status,359):360 organization = make_organization()361 tester = make_user_for_organization(organization, role=role)362 _, token = make_token_for_organization(organization)363 client = APIClient()364 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": tester.public_primary_key})365 with patch(366 "apps.api.views.user.UserView.get_verification_code",367 return_value=Response(368 status=status.HTTP_200_OK,369 ),370 ):371 response = client.get(url, format="json", **make_user_auth_headers(tester, token))372 assert response.status_code == expected_status373@pytest.mark.django_db374@pytest.mark.parametrize(375 "role,expected_status",376 [377 (Role.ADMIN, status.HTTP_200_OK),378 (Role.EDITOR, status.HTTP_403_FORBIDDEN),379 (Role.VIEWER, status.HTTP_403_FORBIDDEN),380 ],381)382def test_user_get_other_verification_code(383 make_organization,384 make_user_for_organization,385 make_token_for_organization,386 make_user_auth_headers,387 role,388 expected_status,389):390 organization = make_organization()391 admin = make_user_for_organization(organization)392 tester = make_user_for_organization(organization, role=role)393 _, token = make_token_for_organization(organization)394 client = APIClient()395 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": admin.public_primary_key})396 with patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock()):397 response = client.get(url, format="json", **make_user_auth_headers(tester, token))398 assert response.status_code == expected_status399@pytest.mark.django_db400@pytest.mark.parametrize(401 "role,expected_status",402 [403 (Role.ADMIN, status.HTTP_200_OK),404 (Role.EDITOR, status.HTTP_200_OK),405 (Role.VIEWER, status.HTTP_403_FORBIDDEN),406 ],407)408def test_user_verify_own_phone(409 make_organization,410 make_user_for_organization,411 make_token_for_organization,412 make_user_auth_headers,413 role,414 expected_status,415):416 organization = make_organization()417 tester = make_user_for_organization(organization, role=role)418 _, token = make_token_for_organization(organization)419 client = APIClient()420 url = reverse("api-internal:user-verify-number", kwargs={"pk": tester.public_primary_key})421 with patch(422 "apps.api.views.user.UserView.verify_number",423 return_value=Response(424 status=status.HTTP_200_OK,425 ),426 ):427 response = client.put(url, format="json", **make_user_auth_headers(tester, token))428 assert response.status_code == expected_status429"""430Tests below are outdated431"""432@pytest.mark.django_db433@pytest.mark.parametrize(434 "role,expected_status",435 [436 (Role.ADMIN, status.HTTP_200_OK),437 (Role.EDITOR, status.HTTP_403_FORBIDDEN),438 (Role.VIEWER, status.HTTP_403_FORBIDDEN),439 ],440)441def test_user_verify_another_phone(442 make_organization,443 make_user_for_organization,444 make_token_for_organization,445 make_user_auth_headers,446 role,447 expected_status,448):449 organization = make_organization()450 tester = make_user_for_organization(organization, role=role)451 other_user = make_user_for_organization(organization, role=Role.EDITOR)452 _, token = make_token_for_organization(organization)453 client = APIClient()454 url = reverse("api-internal:user-verify-number", kwargs={"pk": other_user.public_primary_key})455 with patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None)):456 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(tester, token))457 assert response.status_code == expected_status458@pytest.mark.django_db459@pytest.mark.parametrize(460 "role,expected_status",461 [462 (Role.ADMIN, status.HTTP_200_OK),463 (Role.EDITOR, status.HTTP_200_OK),464 (Role.VIEWER, status.HTTP_403_FORBIDDEN),465 ],466)467def test_user_get_own_telegram_verification_code(468 make_organization,469 make_user_for_organization,470 make_token_for_organization,471 make_user_auth_headers,472 role,473 expected_status,474):475 organization = make_organization()476 tester = make_user_for_organization(organization, role=role)477 _, token = make_token_for_organization(organization)478 client = APIClient()479 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": tester.public_primary_key})480 response = client.get(url, format="json", **make_user_auth_headers(tester, token))481 assert response.status_code == expected_status482@pytest.mark.django_db483@pytest.mark.parametrize(484 "role,expected_status",485 [486 (Role.ADMIN, status.HTTP_200_OK),487 (Role.EDITOR, status.HTTP_403_FORBIDDEN),488 (Role.VIEWER, status.HTTP_403_FORBIDDEN),489 ],490)491def test_user_get_another_telegram_verification_code(492 make_organization,493 make_user_for_organization,494 make_token_for_organization,495 make_user_auth_headers,496 role,497 expected_status,498):499 organization = make_organization()500 tester = make_user_for_organization(organization, role=role)501 other_user = make_user_for_organization(organization, role=Role.EDITOR)502 _, token = make_token_for_organization(organization)503 client = APIClient()504 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": other_user.public_primary_key})505 response = client.get(url, format="json", **make_user_auth_headers(tester, token))506 assert response.status_code == expected_status507@pytest.mark.django_db508def test_admin_can_update_user(509 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers510):511 organization = make_organization()512 tester = make_user_for_organization(organization, role=Role.ADMIN)513 other_user = make_user_for_organization(organization, role=Role.EDITOR)514 _, token = make_token_for_organization(organization)515 client = APIClient()516 data = {517 "email": "test@amixr.io",518 "role": Role.ADMIN,519 "username": "updated_test_username",520 "unverified_phone_number": "+1234567890",521 "slack_login": "",522 }523 url = reverse("api-internal:user-detail", kwargs={"pk": other_user.public_primary_key})524 response = client.put(url, format="json", data=data, **make_user_auth_headers(tester, token))525 assert response.status_code == status.HTTP_200_OK526@pytest.mark.django_db527def test_admin_can_update_himself(528 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers529):530 organization = make_organization()531 admin = make_user_for_organization(organization, role=Role.ADMIN)532 _, token = make_token_for_organization(organization)533 client = APIClient()534 data = {535 "email": "test@amixr.io",536 "role": Role.ADMIN,537 "username": "updated_test_username",538 "unverified_phone_number": "+1234567890",539 "slack_login": "",540 }541 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})542 response = client.put(url, format="json", data=data, **make_user_auth_headers(admin, token))543 assert response.status_code == status.HTTP_200_OK544@pytest.mark.django_db545def test_admin_can_list_users(546 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers547):548 organization = make_organization()549 admin = make_user_for_organization(organization, role=Role.ADMIN)550 _, token = make_token_for_organization(organization)551 client = APIClient()552 url = reverse("api-internal:user-list")553 response = client.get(url, format="json", **make_user_auth_headers(admin, token))554 assert response.status_code == status.HTTP_200_OK555@pytest.mark.django_db556def test_admin_can_detail_users(557 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers558):559 organization = make_organization()560 admin = make_user_for_organization(organization, role=Role.ADMIN)561 editor = make_user_for_organization(organization, role=Role.EDITOR)562 _, token = make_token_for_organization(organization)563 client = APIClient()564 url = reverse("api-internal:user-detail", kwargs={"pk": editor.public_primary_key})565 response = client.get(url, format="json", **make_user_auth_headers(admin, token))566 assert response.status_code == status.HTTP_200_OK567@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())568@pytest.mark.django_db569def test_admin_can_get_own_verification_code(570 mock_verification_start,571 make_organization,572 make_user_for_organization,573 make_token_for_organization,574 make_user_auth_headers,575):576 organization = make_organization()577 admin = make_user_for_organization(organization, role=Role.ADMIN)578 _, token = make_token_for_organization(organization)579 client = APIClient()580 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": admin.public_primary_key})581 response = client.get(url, format="json", **make_user_auth_headers(admin, token))582 assert response.status_code == status.HTTP_200_OK583@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())584@pytest.mark.django_db585def test_admin_can_get_another_user_verification_code(586 mock_verification_start,587 make_organization,588 make_user_for_organization,589 make_token_for_organization,590 make_user_auth_headers,591):592 organization = make_organization()593 admin = make_user_for_organization(organization, role=Role.ADMIN)594 editor = make_user_for_organization(organization, role=Role.EDITOR)595 _, token = make_token_for_organization(organization)596 client = APIClient()597 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": editor.public_primary_key})598 response = client.get(url, format="json", **make_user_auth_headers(admin, token))599 assert response.status_code == status.HTTP_200_OK600@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))601@pytest.mark.django_db602def test_admin_can_verify_own_phone(603 mocked_verification_check,604 make_organization,605 make_user_for_organization,606 make_token_for_organization,607 make_user_auth_headers,608):609 organization = make_organization()610 admin = make_user_for_organization(organization, role=Role.ADMIN)611 _, token = make_token_for_organization(organization)612 client = APIClient()613 url = reverse("api-internal:user-verify-number", kwargs={"pk": admin.public_primary_key})614 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(admin, token))615 assert response.status_code == status.HTTP_200_OK616@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))617@pytest.mark.django_db618def test_admin_can_verify_another_user_phone(619 mocked_verification_check,620 make_organization,621 make_user_for_organization,622 make_token_for_organization,623 make_user_auth_headers,624):625 organization = make_organization()626 admin = make_user_for_organization(organization, role=Role.ADMIN)627 editor = make_user_for_organization(organization, role=Role.EDITOR)628 _, token = make_token_for_organization(organization)629 client = APIClient()630 url = reverse("api-internal:user-verify-number", kwargs={"pk": editor.public_primary_key})631 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(admin, token))632 assert response.status_code == status.HTTP_200_OK633@pytest.mark.django_db634def test_admin_can_get_own_telegram_verification_code(635 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers636):637 organization = make_organization()638 admin = make_user_for_organization(organization, role=Role.ADMIN)639 _, token = make_token_for_organization(organization)640 client = APIClient()641 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": admin.public_primary_key})642 response = client.get(url, format="json", **make_user_auth_headers(admin, token))643 assert response.status_code == status.HTTP_200_OK644@pytest.mark.django_db645def test_admin_can_get_another_user_telegram_verification_code(646 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers647):648 organization = make_organization()649 admin = make_user_for_organization(organization, role=Role.ADMIN)650 editor = make_user_for_organization(organization, role=Role.EDITOR)651 _, token = make_token_for_organization(organization)652 client = APIClient()653 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": editor.public_primary_key})654 response = client.get(url, format="json", **make_user_auth_headers(admin, token))655 assert response.status_code == status.HTTP_200_OK656@pytest.mark.django_db657def test_admin_can_get_another_user_backend_verification_code(658 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers659):660 organization = make_organization()661 admin = make_user_for_organization(organization, role=Role.ADMIN)662 editor = make_user_for_organization(organization, role=Role.EDITOR)663 _, token = make_token_for_organization(organization)664 client = APIClient()665 url = (666 reverse("api-internal:user-get-backend-verification-code", kwargs={"pk": editor.public_primary_key})667 + "?backend=TESTONLY"668 )669 response = client.get(url, format="json", **make_user_auth_headers(admin, token))670 assert response.status_code == status.HTTP_200_OK671@pytest.mark.django_db672def test_admin_can_unlink_another_user_backend_account(673 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers674):675 organization = make_organization()676 admin = make_user_for_organization(organization, role=Role.ADMIN)677 editor = make_user_for_organization(organization, role=Role.EDITOR)678 _, token = make_token_for_organization(organization)679 client = APIClient()680 url = reverse("api-internal:user-unlink-backend", kwargs={"pk": editor.public_primary_key}) + "?backend=TESTONLY"681 response = client.post(url, format="json", **make_user_auth_headers(admin, token))682 assert response.status_code == status.HTTP_200_OK683@pytest.mark.django_db684def test_admin_can_unlink_another_user_slack_account(685 make_organization_with_slack_team_identity,686 make_user_for_organization,687 make_user_with_slack_user_identity,688 make_token_for_organization,689 make_user_auth_headers,690):691 organization, slack_team_identity = make_organization_with_slack_team_identity()692 admin = make_user_for_organization(organization, role=Role.ADMIN)693 editor, slack_user_identity_1 = make_user_with_slack_user_identity(694 slack_team_identity, organization, slack_id="user_1", role=Role.EDITOR695 )696 _, token = make_token_for_organization(organization)697 client = APIClient()698 url = reverse("api-internal:user-unlink-slack", kwargs={"pk": editor.public_primary_key})699 response = client.post(url, format="json", **make_user_auth_headers(admin, token))700 assert response.status_code == status.HTTP_200_OK701 editor.refresh_from_db()702 assert editor.slack_user_identity is None703"""Test user permissions"""704@pytest.mark.django_db705def test_user_cant_update_user(706 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers707):708 organization = make_organization()709 first_user = make_user_for_organization(organization, role=Role.EDITOR)710 second_user = make_user_for_organization(organization, role=Role.EDITOR)711 _, token = make_token_for_organization(organization)712 client = APIClient()713 data = {714 "email": "test@amixr.io",715 "role": Role.ADMIN,716 "username": "updated_test_username",717 "unverified_phone_number": "+1234567890",718 "slack_login": "",719 }720 url = reverse("api-internal:user-detail", kwargs={"pk": first_user.public_primary_key})721 response = client.put(url, format="json", data=data, **make_user_auth_headers(second_user, token))722 assert response.status_code == status.HTTP_403_FORBIDDEN723@pytest.mark.django_db724def test_user_can_update_themself(725 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers726):727 organization = make_organization()728 user = make_user_for_organization(organization, role=Role.EDITOR)729 _, token = make_token_for_organization(organization)730 client = APIClient()731 data = {732 "email": "test@amixr.io",733 "role": Role.EDITOR,734 "username": "updated_test_username",735 "unverified_phone_number": "+1234567890",736 "slack_login": "",737 }738 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})739 response = client.put(url, format="json", data=data, **make_user_auth_headers(user, token))740 assert response.status_code == status.HTTP_200_OK741@pytest.mark.django_db742def test_user_can_list_users(743 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers744):745 organization = make_organization()746 editor = make_user_for_organization(organization, role=Role.EDITOR)747 _, token = make_token_for_organization(organization)748 client = APIClient()749 url = reverse("api-internal:user-list")750 response = client.get(url, format="json", **make_user_auth_headers(editor, token))751 assert response.status_code == status.HTTP_200_OK752@pytest.mark.django_db753def test_user_can_detail_users(754 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers755):756 organization = make_organization()757 admin = make_user_for_organization(organization, role=Role.ADMIN)758 editor = make_user_for_organization(organization, role=Role.EDITOR)759 _, token = make_token_for_organization(organization)760 client = APIClient()761 url = reverse("api-internal:user-detail", kwargs={"pk": admin.public_primary_key})762 response = client.get(url, format="json", **make_user_auth_headers(editor, token))763 assert response.status_code == status.HTTP_403_FORBIDDEN764@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())765@pytest.mark.django_db766def test_user_can_get_own_verification_code(767 mock_verification_start,768 make_organization,769 make_user_for_organization,770 make_token_for_organization,771 make_user_auth_headers,772):773 organization = make_organization()774 user = make_user_for_organization(organization, role=Role.EDITOR)775 _, token = make_token_for_organization(organization)776 client = APIClient()777 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": user.public_primary_key})778 response = client.get(f"{url}", format="json", **make_user_auth_headers(user, token))779 assert response.status_code == status.HTTP_200_OK780@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())781@pytest.mark.django_db782def test_user_cant_get_another_user_verification_code(783 mock_verification_start,784 make_organization,785 make_user_for_organization,786 make_token_for_organization,787 make_user_auth_headers,788):789 organization = make_organization()790 first_user = make_user_for_organization(organization, role=Role.EDITOR)791 second_user = make_user_for_organization(organization, role=Role.EDITOR)792 _, token = make_token_for_organization(organization)793 client = APIClient()794 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": first_user.public_primary_key})795 response = client.get(f"{url}", format="json", **make_user_auth_headers(second_user, token))796 assert response.status_code == status.HTTP_403_FORBIDDEN797@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))798@pytest.mark.django_db799def test_user_can_verify_own_phone(800 mocked_verification_check,801 make_organization,802 make_user_for_organization,803 make_token_for_organization,804 make_user_auth_headers,805):806 organization = make_organization()807 user = make_user_for_organization(organization, role=Role.EDITOR)808 _, token = make_token_for_organization(organization)809 client = APIClient()810 url = reverse("api-internal:user-verify-number", kwargs={"pk": user.public_primary_key})811 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(user, token))812 assert response.status_code == status.HTTP_200_OK813@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))814@pytest.mark.django_db815def test_user_cant_verify_another_user_phone(816 mocked_verification_check,817 make_organization,818 make_user_for_organization,819 make_token_for_organization,820 make_user_auth_headers,821):822 organization = make_organization()823 first_user = make_user_for_organization(organization, role=Role.EDITOR)824 second_user = make_user_for_organization(organization, role=Role.EDITOR)825 _, token = make_token_for_organization(organization)826 client = APIClient()827 url = reverse("api-internal:user-verify-number", kwargs={"pk": first_user.public_primary_key})828 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(second_user, token))829 assert response.status_code == status.HTTP_403_FORBIDDEN830@pytest.mark.django_db831def test_user_can_get_own_telegram_verification_code(832 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers833):834 organization = make_organization()835 user = make_user_for_organization(organization, role=Role.EDITOR)836 _, token = make_token_for_organization(organization)837 client = APIClient()838 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": user.public_primary_key})839 response = client.get(f"{url}", format="json", **make_user_auth_headers(user, token))840 assert response.status_code == status.HTTP_200_OK841@pytest.mark.django_db842def test_user_cant_get_another_user_telegram_verification_code(843 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers844):845 organization = make_organization()846 first_user = make_user_for_organization(organization, role=Role.EDITOR)847 second_user = make_user_for_organization(organization, role=Role.EDITOR)848 _, token = make_token_for_organization(organization)849 client = APIClient()850 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": first_user.public_primary_key})851 response = client.get(f"{url}", format="json", **make_user_auth_headers(second_user, token))852 assert response.status_code == status.HTTP_403_FORBIDDEN853@pytest.mark.django_db854def test_user_can_get_own_backend_verification_code(855 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers856):857 organization = make_organization()858 user = make_user_for_organization(organization, role=Role.EDITOR)859 _, token = make_token_for_organization(organization)860 client = APIClient()861 url = (862 reverse("api-internal:user-get-backend-verification-code", kwargs={"pk": user.public_primary_key})863 + "?backend=TESTONLY"864 )865 with patch(866 "apps.base.tests.messaging_backend.TestOnlyBackend.generate_user_verification_code",867 return_value="the-code",868 ) as mock_generate_code:869 response = client.get(f"{url}", format="json", **make_user_auth_headers(user, token))870 assert response.status_code == status.HTTP_200_OK871 assert response.json() == "the-code"872 mock_generate_code.assert_called_once_with(user)873@pytest.mark.django_db874def test_user_cant_get_another_user_backend_verification_code(875 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers876):877 organization = make_organization()878 first_user = make_user_for_organization(organization, role=Role.EDITOR)879 second_user = make_user_for_organization(organization, role=Role.EDITOR)880 _, token = make_token_for_organization(organization)881 client = APIClient()882 url = (883 reverse("api-internal:user-get-backend-verification-code", kwargs={"pk": first_user.public_primary_key})884 + "?backend=TESTONLY"885 )886 response = client.get(f"{url}", format="json", **make_user_auth_headers(second_user, token))887 assert response.status_code == status.HTTP_403_FORBIDDEN888@pytest.mark.django_db889def test_user_can_unlink_own_slack_account(890 make_organization_with_slack_team_identity,891 make_user_with_slack_user_identity,892 make_token_for_organization,893 make_user_auth_headers,894):895 organization, slack_team_identity = make_organization_with_slack_team_identity()896 user, slack_user_identity_1 = make_user_with_slack_user_identity(897 slack_team_identity, organization, slack_id="user_1", role=Role.EDITOR898 )899 _, token = make_token_for_organization(organization)900 client = APIClient()901 url = reverse("api-internal:user-unlink-slack", kwargs={"pk": user.public_primary_key})902 response = client.post(url, format="json", **make_user_auth_headers(user, token))903 assert response.status_code == status.HTTP_200_OK904 user.refresh_from_db()905 assert user.slack_user_identity is None906@pytest.mark.django_db907def test_user_can_unlink_backend_own_account(908 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers909):910 organization = make_organization()911 user = make_user_for_organization(organization, role=Role.EDITOR)912 _, token = make_token_for_organization(organization)913 client = APIClient()914 url = reverse("api-internal:user-unlink-backend", kwargs={"pk": user.public_primary_key}) + "?backend=TESTONLY"915 response = client.post(f"{url}", format="json", **make_user_auth_headers(user, token))916 assert response.status_code == status.HTTP_200_OK917@pytest.mark.django_db918def test_user_unlink_backend_invalid_backend_id(919 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers920):921 organization = make_organization()922 user = make_user_for_organization(organization, role=Role.EDITOR)923 _, token = make_token_for_organization(organization)924 client = APIClient()925 url = reverse("api-internal:user-unlink-backend", kwargs={"pk": user.public_primary_key}) + "?backend=INVALID"926 response = client.post(f"{url}", format="json", **make_user_auth_headers(user, token))927 assert response.status_code == status.HTTP_400_BAD_REQUEST928@pytest.mark.django_db929def test_user_unlink_backend_backend_account_not_found(930 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers931):932 organization = make_organization()933 user = make_user_for_organization(organization, role=Role.EDITOR)934 _, token = make_token_for_organization(organization)935 client = APIClient()936 url = reverse("api-internal:user-unlink-backend", kwargs={"pk": user.public_primary_key}) + "?backend=TESTONLY"937 with patch("apps.base.tests.messaging_backend.TestOnlyBackend.unlink_user", side_effect=ObjectDoesNotExist):938 response = client.post(f"{url}", format="json", **make_user_auth_headers(user, token))939 assert response.status_code == status.HTTP_400_BAD_REQUEST940@pytest.mark.django_db941def test_user_cant_unlink_slack_another_user(942 make_organization_with_slack_team_identity,943 make_user_with_slack_user_identity,944 make_token_for_organization,945 make_user_auth_headers,946):947 organization, slack_team_identity = make_organization_with_slack_team_identity()948 first_user, slack_user_identity_1 = make_user_with_slack_user_identity(949 slack_team_identity, organization, slack_id="user_1", role=Role.EDITOR950 )951 second_user, slack_user_identity_2 = make_user_with_slack_user_identity(952 slack_team_identity, organization, slack_id="user_2", role=Role.EDITOR953 )954 _, token = make_token_for_organization(organization)955 client = APIClient()956 url = reverse("api-internal:user-unlink-slack", kwargs={"pk": first_user.public_primary_key})957 response = client.post(url, format="json", **make_user_auth_headers(second_user, token))958 assert response.status_code == status.HTTP_403_FORBIDDEN959 first_user.refresh_from_db()960 assert first_user.slack_user_identity is not None961@pytest.mark.django_db962def test_user_cant_unlink_backend__another_user(963 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers964):965 organization = make_organization()966 first_user = make_user_for_organization(organization, role=Role.EDITOR)967 second_user = make_user_for_organization(organization, role=Role.EDITOR)968 _, token = make_token_for_organization(organization)969 client = APIClient()970 url = (971 reverse("api-internal:user-unlink-backend", kwargs={"pk": first_user.public_primary_key}) + "?backend=TESTONLY"972 )973 response = client.post(f"{url}", format="json", **make_user_auth_headers(second_user, token))974 assert response.status_code == status.HTTP_403_FORBIDDEN975"""Test stakeholder permissions"""976@pytest.mark.django_db977def test_viewer_cant_create_user(978 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers979):980 organization = make_organization()981 user = make_user_for_organization(organization, role=Role.VIEWER)982 _, token = make_token_for_organization(organization)983 client = APIClient()984 url = reverse("api-internal:user-list")985 data = {986 "email": "test@amixr.io",987 "role": Role.ADMIN,988 "username": "test_username",989 "unverified_phone_number": None,990 "slack_login": "",991 }992 response = client.post(url, format="json", data=data, **make_user_auth_headers(user, token))993 assert response.status_code == status.HTTP_403_FORBIDDEN994@pytest.mark.django_db995def test_viewer_cant_update_user(996 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers997):998 organization = make_organization()999 first_user = make_user_for_organization(organization, role=Role.EDITOR)1000 second_user = make_user_for_organization(organization, role=Role.VIEWER)1001 _, token = make_token_for_organization(organization)1002 data = {1003 "email": "test@amixr.io",1004 "role": Role.EDITOR,1005 "username": "updated_test_username",1006 "unverified_phone_number": "+1234567890",1007 "slack_login": "",1008 }1009 client = APIClient()1010 url = reverse("api-internal:user-detail", kwargs={"pk": first_user.public_primary_key})1011 response = client.put(url, format="json", data=data, **make_user_auth_headers(second_user, token))1012 assert response.status_code == status.HTTP_403_FORBIDDEN1013@pytest.mark.django_db1014def test_viewer_cant_update_himself(1015 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1016):1017 organization = make_organization()1018 user = make_user_for_organization(organization, role=Role.VIEWER)1019 _, token = make_token_for_organization(organization)1020 data = {1021 "email": "test@amixr.io",1022 "role": Role.VIEWER,1023 "username": "updated_test_username",1024 "unverified_phone_number": "+1234567890",1025 "slack_login": "",1026 }1027 client = APIClient()1028 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1029 response = client.put(url, format="json", data=data, **make_user_auth_headers(user, token))1030 assert response.status_code == status.HTTP_403_FORBIDDEN1031@pytest.mark.django_db1032def test_viewer_cant_list_users(1033 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1034):1035 organization = make_organization()1036 user = make_user_for_organization(organization, role=Role.VIEWER)1037 _, token = make_token_for_organization(organization)1038 client = APIClient()1039 url = reverse("api-internal:user-list")1040 response = client.get(url, format="json", **make_user_auth_headers(user, token))1041 assert response.status_code == status.HTTP_403_FORBIDDEN1042@pytest.mark.django_db1043def test_viewer_cant_detail_users(1044 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1045):1046 organization = make_organization()1047 first_user = make_user_for_organization(organization, role=Role.EDITOR)1048 second_user = make_user_for_organization(organization, role=Role.VIEWER)1049 _, token = make_token_for_organization(organization)1050 client = APIClient()1051 url = reverse("api-internal:user-detail", kwargs={"pk": first_user.public_primary_key})1052 response = client.get(url, format="json", **make_user_auth_headers(second_user, token))1053 assert response.status_code == status.HTTP_403_FORBIDDEN1054@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())1055@pytest.mark.django_db1056def test_viewer_cant_get_own_verification_code(1057 mock_verification_start,1058 make_organization,1059 make_user_for_organization,1060 make_token_for_organization,1061 make_user_auth_headers,1062):1063 organization = make_organization()1064 user = make_user_for_organization(organization, role=Role.VIEWER)1065 _, token = make_token_for_organization(organization)1066 client = APIClient()1067 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": user.public_primary_key})1068 response = client.get(url, format="json", **make_user_auth_headers(user, token))1069 assert response.status_code == status.HTTP_403_FORBIDDEN1070@patch("apps.twilioapp.phone_manager.PhoneManager.send_verification_code", return_value=Mock())1071@pytest.mark.django_db1072def test_viewer_cant_get_another_user_verification_code(1073 mock_verification_start,1074 make_organization,1075 make_user_for_organization,1076 make_token_for_organization,1077 make_user_auth_headers,1078):1079 organization = make_organization()1080 first_user = make_user_for_organization(organization, role=Role.EDITOR)1081 second_user = make_user_for_organization(organization, role=Role.VIEWER)1082 _, token = make_token_for_organization(organization)1083 client = APIClient()1084 url = reverse("api-internal:user-get-verification-code", kwargs={"pk": first_user.public_primary_key})1085 response = client.get(url, format="json", **make_user_auth_headers(second_user, token))1086 assert response.status_code == status.HTTP_403_FORBIDDEN1087@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))1088@pytest.mark.django_db1089def test_viewer_cant_verify_own_phone(1090 mocked_verification_check,1091 make_organization,1092 make_user_for_organization,1093 make_token_for_organization,1094 make_user_auth_headers,1095):1096 organization = make_organization()1097 user = make_user_for_organization(organization, role=Role.VIEWER)1098 _, token = make_token_for_organization(organization)1099 client = APIClient()1100 url = reverse("api-internal:user-verify-number", kwargs={"pk": user.public_primary_key})1101 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(user, token))1102 assert response.status_code == status.HTTP_403_FORBIDDEN1103@patch("apps.twilioapp.phone_manager.PhoneManager.verify_phone_number", return_value=(True, None))1104@pytest.mark.django_db1105def test_viewer_cant_verify_another_user_phone(1106 mocked_verification_check,1107 make_organization,1108 make_user_for_organization,1109 make_token_for_organization,1110 make_user_auth_headers,1111):1112 organization = make_organization()1113 first_user = make_user_for_organization(organization, role=Role.EDITOR)1114 second_user = make_user_for_organization(organization, role=Role.VIEWER)1115 _, token = make_token_for_organization(organization)1116 client = APIClient()1117 url = reverse("api-internal:user-verify-number", kwargs={"pk": first_user.public_primary_key})1118 response = client.put(f"{url}?token=12345", format="json", **make_user_auth_headers(second_user, token))1119 assert response.status_code == status.HTTP_403_FORBIDDEN1120@pytest.mark.django_db1121def test_viewer_cant_get_own_telegram_verification_code(1122 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1123):1124 organization = make_organization()1125 user = make_user_for_organization(organization, role=Role.VIEWER)1126 _, token = make_token_for_organization(organization)1127 client = APIClient()1128 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": user.public_primary_key})1129 response = client.get(f"{url}", format="json", **make_user_auth_headers(user, token))1130 assert response.status_code == status.HTTP_403_FORBIDDEN1131@pytest.mark.django_db1132def test_viewer_cant_get_another_user_telegram_verification_code(1133 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1134):1135 organization = make_organization()1136 first_user = make_user_for_organization(organization, role=Role.EDITOR)1137 second_user = make_user_for_organization(organization, role=Role.VIEWER)1138 _, token = make_token_for_organization(organization)1139 client = APIClient()1140 url = reverse("api-internal:user-get-telegram-verification-code", kwargs={"pk": first_user.public_primary_key})1141 response = client.get(url, format="json", **make_user_auth_headers(second_user, token))1142 assert response.status_code == status.HTTP_403_FORBIDDEN1143@pytest.mark.django_db1144@pytest.mark.parametrize(1145 "role,expected_status,initial_unverified_number,initial_verified_number",1146 [1147 (Role.ADMIN, status.HTTP_200_OK, "+1234567890", None),1148 (Role.EDITOR, status.HTTP_200_OK, "+1234567890", None),1149 (Role.VIEWER, status.HTTP_403_FORBIDDEN, "+1234567890", None),1150 (Role.ADMIN, status.HTTP_200_OK, None, "+1234567890"),1151 (Role.EDITOR, status.HTTP_200_OK, None, "+1234567890"),1152 (Role.VIEWER, status.HTTP_403_FORBIDDEN, None, "+1234567890"),1153 ],1154)1155def test_forget_own_number(1156 make_organization,1157 make_team,1158 make_user_for_organization,1159 make_token_for_organization,1160 make_user_auth_headers,1161 role,1162 expected_status,1163 initial_unverified_number,1164 initial_verified_number,1165):1166 organization = make_organization()1167 admin = make_user_for_organization(organization, role=Role.ADMIN)1168 user = make_user_for_organization(1169 organization,1170 role=role,1171 unverified_phone_number=initial_unverified_number,1172 _verified_phone_number=initial_verified_number,1173 )1174 _, token = make_token_for_organization(organization)1175 client = APIClient()1176 url = reverse("api-internal:user-forget-number", kwargs={"pk": user.public_primary_key})1177 with patch(1178 "apps.twilioapp.phone_manager.PhoneManager.notify_about_changed_verified_phone_number", return_value=None1179 ):1180 response = client.put(url, None, format="json", **make_user_auth_headers(user, token))1181 assert response.status_code == expected_status1182 user_detail_url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1183 response = client.get(user_detail_url, None, format="json", **make_user_auth_headers(admin, token))1184 assert response.status_code == status.HTTP_200_OK1185 if expected_status == status.HTTP_200_OK:1186 assert not response.json()["unverified_phone_number"]1187 assert not response.json()["verified_phone_number"]1188 else:1189 assert response.json()["unverified_phone_number"] == initial_unverified_number1190 assert response.json()["verified_phone_number"] == initial_verified_number1191@pytest.mark.django_db1192@pytest.mark.parametrize(1193 "role,expected_status,initial_unverified_number,initial_verified_number",1194 [1195 (Role.ADMIN, status.HTTP_200_OK, "+1234567890", None),1196 (Role.EDITOR, status.HTTP_403_FORBIDDEN, "+1234567890", None),1197 (Role.VIEWER, status.HTTP_403_FORBIDDEN, "+1234567890", None),1198 (Role.ADMIN, status.HTTP_200_OK, None, "+1234567890"),1199 (Role.EDITOR, status.HTTP_403_FORBIDDEN, None, "+1234567890"),1200 (Role.VIEWER, status.HTTP_403_FORBIDDEN, None, "+1234567890"),1201 ],1202)1203def test_forget_other_number(1204 make_organization,1205 make_team,1206 make_user_for_organization,1207 make_token_for_organization,1208 make_user_auth_headers,1209 role,1210 expected_status,1211 initial_unverified_number,1212 initial_verified_number,1213):1214 organization = make_organization()1215 user = make_user_for_organization(1216 organization,1217 role=Role.ADMIN,1218 unverified_phone_number=initial_unverified_number,1219 _verified_phone_number=initial_verified_number,1220 )1221 other_user = make_user_for_organization(organization, role=role)1222 _, token = make_token_for_organization(organization)1223 client = APIClient()1224 url = reverse("api-internal:user-forget-number", kwargs={"pk": user.public_primary_key})1225 with patch(1226 "apps.twilioapp.phone_manager.PhoneManager.notify_about_changed_verified_phone_number", return_value=None1227 ):1228 response = client.put(url, None, format="json", **make_user_auth_headers(other_user, token))1229 assert response.status_code == expected_status1230 user_detail_url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1231 response = client.get(user_detail_url, None, format="json", **make_user_auth_headers(user, token))1232 assert response.status_code == status.HTTP_200_OK1233 if expected_status == status.HTTP_200_OK:1234 assert not response.json()["unverified_phone_number"]1235 assert not response.json()["verified_phone_number"]1236 else:1237 assert response.json()["unverified_phone_number"] == initial_unverified_number1238 assert response.json()["verified_phone_number"] == initial_verified_number1239@pytest.mark.django_db1240def test_viewer_cant_get_own_backend_verification_code(1241 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1242):1243 organization = make_organization()1244 user = make_user_for_organization(organization, role=Role.VIEWER)1245 _, token = make_token_for_organization(organization)1246 client = APIClient()1247 url = (1248 reverse("api-internal:user-get-backend-verification-code", kwargs={"pk": user.public_primary_key})1249 + "?backend=TESTONLY"1250 )1251 response = client.get(f"{url}", format="json", **make_user_auth_headers(user, token))1252 assert response.status_code == status.HTTP_403_FORBIDDEN1253@pytest.mark.django_db1254def test_viewer_cant_get_another_user_backend_verification_code(1255 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1256):1257 organization = make_organization()1258 first_user = make_user_for_organization(organization, role=Role.EDITOR)1259 second_user = make_user_for_organization(organization, role=Role.VIEWER)1260 _, token = make_token_for_organization(organization)1261 client = APIClient()1262 url = (1263 reverse("api-internal:user-get-backend-verification-code", kwargs={"pk": first_user.public_primary_key})1264 + "?backend=TESTONLY"1265 )1266 response = client.get(url, format="json", **make_user_auth_headers(second_user, token))1267 assert response.status_code == status.HTTP_403_FORBIDDEN1268@pytest.mark.django_db1269def test_viewer_cant_unlink_backend_own_user(1270 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1271):1272 organization = make_organization()1273 user = make_user_for_organization(organization, role=Role.VIEWER)1274 _, token = make_token_for_organization(organization)1275 client = APIClient()1276 url = reverse("api-internal:user-unlink-backend", kwargs={"pk": user.public_primary_key}) + "?backend=TESTONLY"1277 response = client.post(f"{url}", format="json", **make_user_auth_headers(user, token))1278 assert response.status_code == status.HTTP_403_FORBIDDEN1279@pytest.mark.django_db1280def test_viewer_cant_unlink_backend_another_user(1281 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1282):1283 organization = make_organization()1284 first_user = make_user_for_organization(organization, role=Role.EDITOR)1285 second_user = make_user_for_organization(organization, role=Role.VIEWER)1286 _, token = make_token_for_organization(organization)1287 client = APIClient()1288 url = (1289 reverse("api-internal:user-unlink-backend", kwargs={"pk": first_user.public_primary_key}) + "?backend=TESTONLY"1290 )1291 response = client.post(url, format="json", **make_user_auth_headers(second_user, token))1292 assert response.status_code == status.HTTP_403_FORBIDDEN1293@pytest.mark.django_db1294def test_change_timezone(1295 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1296):1297 organization = make_organization()1298 user = make_user_for_organization(organization, role=Role.EDITOR)1299 _, token = make_token_for_organization(organization)1300 client = APIClient()1301 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1302 data = {"timezone": "Europe/​London"}1303 response = client.put(f"{url}", data, format="json", **make_user_auth_headers(user, token))1304 assert response.status_code == status.HTTP_200_OK1305 assert "timezone" in response.json()1306 assert response.json()["timezone"] == "Europe/​London"1307@pytest.mark.django_db1308@pytest.mark.parametrize("timezone", ["", 1, "NotATimezone"])1309def test_invalid_timezone(1310 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers, timezone1311):1312 organization = make_organization()1313 user = make_user_for_organization(organization, role=Role.EDITOR)1314 _, token = make_token_for_organization(organization)1315 client = APIClient()1316 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1317 data = {"timezone": timezone}1318 response = client.put(f"{url}", data, format="json", **make_user_auth_headers(user, token))1319 assert response.status_code == status.HTTP_400_BAD_REQUEST1320@pytest.mark.django_db1321def test_change_working_hours(1322 make_organization, make_user_for_organization, make_token_for_organization, make_user_auth_headers1323):1324 organization = make_organization()1325 user = make_user_for_organization(organization, role=Role.EDITOR)1326 _, token = make_token_for_organization(organization)1327 client = APIClient()1328 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1329 periods = [{"start": "05:00:00", "end": "23:00:00"}]1330 working_hours = {1331 day: periods for day in ["monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"]1332 }1333 data = {"working_hours": working_hours}1334 response = client.put(f"{url}", data, format="json", **make_user_auth_headers(user, token))1335 assert response.status_code == status.HTTP_200_OK1336 assert "working_hours" in response.json()1337 assert response.json()["working_hours"] == working_hours1338@pytest.mark.django_db1339@pytest.mark.parametrize(1340 "working_hours_extra",1341 [1342 {},1343 {"sunday": 1},1344 {"sunday": ""},1345 {"sunday": {"start": "18:00:00"}},1346 {"sunday": {"start": "", "end": ""}},1347 {"sunday": {"start": "18:00:00", "end": None}},1348 {"sunday": {"start": "18:00:00", "end": "18:00:00"}},1349 {"sunday": {"start": "18:00:00", "end": "9:00:00"}},1350 {"sunday": {"start": "18:00:00", "end": "9:00:00", "extra": 1}},1351 ],1352)1353def test_invalid_working_hours(1354 make_organization,1355 make_user_for_organization,1356 make_token_for_organization,1357 make_user_auth_headers,1358 working_hours_extra,1359):1360 organization = make_organization()1361 user = make_user_for_organization(organization, role=Role.EDITOR)1362 _, token = make_token_for_organization(organization)1363 client = APIClient()1364 url = reverse("api-internal:user-detail", kwargs={"pk": user.public_primary_key})1365 periods = [{"start": "05:00:00", "end": "23:00:00"}]1366 working_hours = {day: periods for day in ["monday", "tuesday", "wednesday", "thursday", "friday", "saturday"]}1367 working_hours.update(working_hours_extra)1368 data = {"working_hours": working_hours}1369 response = client.put(f"{url}", data, format="json", **make_user_auth_headers(user, token))...

Full Screen

Full Screen

make_einstein.py

Source:make_einstein.py Github

copy

Full Screen

1from enum import Enum2class Nationality(Enum):3 brit = 04 swede = 15 dane = 26 norwegian = 37 german = 48 def __int__(self):9 return self.value10class Color(Enum):11 red = 012 green = 113 yellow = 214 blue = 315 white = 416 def __int__(self):17 return self.value18class Beverage(Enum):19 tea = 020 coffee = 121 milk = 222 beer = 323 water = 424 def __int__(self):25 return self.value26class Cigar(Enum):27 pallmall = 028 dunhill = 129 blends = 230 bluemasters = 331 prince = 432 def __int__(self):33 return self.value34class Pet(Enum):35 dogs = 036 birds = 137 cats = 238 horse = 339 fish = 440 def __int__(self):41 return self.value42class Clause:43 def __init__(self):44 self.s = set([])45 def __str__(self):46 ret_str = ""47 for p in self.s:48 ret_str += str(p)49 ret_str += " "50 ret_str += "0"51 return ret_str52 53 def __repr__(self):54 ret_str = ""55 for p in self.s:56 ret_str += str(p)57 ret_str += " "58 ret_str += "0"59 return ret_str60 def add_literal(self, l):61 self.s.add(l)62def make_literal(prop, negated=False):63 if negated:64 return -(prop + 1)65 else:66 return prop + 167 68def make_einstein_prop(n=None, c=None, b=None, s=None, p=None, h=None):69 assert(isinstance(n, Nationality) or n is None)70 assert(isinstance(c, Color) or c is None)71 assert(isinstance(b, Beverage) or b is None)72 assert(isinstance(s, Cigar) or s is None)73 assert(isinstance(p, Pet) or p is None)74 assert(isinstance(h, int) or h is None)75 num_not_none = sum([x is not None for x in [n, c, b, s, p, h]])76 assert(num_not_none==2)77 prop_num = 078 # Nationality79 if n is not None:80 if c is not None:81 # NC82 prop_num = prop_num + (int(n)*5) + int(c)83 return prop_num84 elif b is not None:85 # NB86 prop_num = 2587 prop_num = prop_num + (int(n)*5) + int(b)88 return prop_num89 elif s is not None:90 # NS91 prop_num = 5092 prop_num = prop_num + (int(n)*5) + int(s)93 return prop_num94 elif p is not None:95 # NP96 prop_num = 7597 prop_num = prop_num + (int(n)*5) + int(p)98 return prop_num99 else:100 # NH101 prop_num = 100102 prop_num = prop_num + (int(n)*5) + int(h)103 return prop_num104 # Color105 elif c is not None:106 prop_num = 125107 if b is not None:108 # CB109 prop_num = prop_num + (int(c)*5) + int(b)110 return prop_num111 elif s is not None:112 # CS113 prop_num = prop_num + 25114 prop_num = prop_num + (int(c)*5) + int(s)115 return prop_num116 elif p is not None:117 # CP118 prop_num = prop_num + 50119 prop_num = prop_num + (int(c)*5) + int(p)120 return prop_num121 else:122 # CH123 prop_num = prop_num + 75124 prop_num = prop_num + (int(c)*5) + int(h)125 return prop_num126 # Beverage127 elif b is not None:128 prop_num = 225129 if s is not None:130 # BS131 prop_num = prop_num + (int(b)*5) + int(s)132 return prop_num133 elif p is not None:134 # BP135 prop_num = prop_num + 25136 prop_num = prop_num + (int(b)*5) + int(p)137 return prop_num138 else:139 # BH140 prop_num = prop_num + 50141 prop_num = prop_num + (int(b)*5) + int(h)142 return prop_num143 # Smoking144 elif s is not None:145 prop_num = 300146 if p is not None:147 # SP148 prop_num = prop_num + (int(s)*5) + int(p)149 return prop_num150 else:151 # SH152 prop_num = prop_num + 25153 prop_num = prop_num + (int(s)*5) + int(h)154 return prop_num155 # Pet156 else:157 # PH158 prop_num = 350159 prop_num = prop_num + (int(p)*5) + int(h)160 return prop_num161def make_exclusion_axioms(ns=None, cs=None, bs=None, ss=None, ps=None, hs=None):162 num_not_none = sum([x is not None for x in [ns, cs, bs, ss, ps, hs]])163 assert(num_not_none==2)164 axioms = []165 # Nationality166 if ns is not None:167 if cs is not None:168 # NC169 for c in cs:170 disjunctive_c = Clause()171 for i in range(len(ns)):172 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], c=c)))173 for j in range(i+1, len(ns)):174 exclusive_c = Clause()175 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], c=c), negated=True))176 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[j], c=c), negated=True))177 axioms.append(exclusive_c)178 axioms.append(disjunctive_c)179 for n in ns:180 disjunctive_c = Clause()181 for i in range(len(cs)):182 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=n, c=cs[i])))183 for j in range(i+1, len(cs)):184 exclusive_c = Clause()185 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, c=cs[i]), negated=True))186 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, c=cs[j]), negated=True))187 axioms.append(exclusive_c)188 axioms.append(disjunctive_c)189 elif bs is not None:190 # NB191 for b in bs:192 disjunctive_c = Clause()193 for i in range(len(ns)):194 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], b=b)))195 for j in range(i+1, len(ns)):196 exclusive_c = Clause()197 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], b=b), negated=True))198 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[j], b=b), negated=True))199 axioms.append(exclusive_c)200 axioms.append(disjunctive_c)201 for n in ns:202 disjunctive_c = Clause()203 for i in range(len(bs)):204 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=n, b=bs[i])))205 for j in range(i+1, len(bs)):206 exclusive_c = Clause()207 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, b=bs[i]), negated=True))208 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, b=bs[j]), negated=True))209 axioms.append(exclusive_c)210 axioms.append(disjunctive_c)211 elif ss is not None:212 # NS213 for s in ss:214 disjunctive_c = Clause()215 for i in range(len(ns)):216 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], s=s)))217 for j in range(i+1, len(ns)):218 exclusive_c = Clause()219 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], s=s), negated=True))220 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[j], s=s), negated=True))221 axioms.append(exclusive_c)222 axioms.append(disjunctive_c)223 for n in ns:224 disjunctive_c = Clause()225 for i in range(len(ss)):226 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=n, s=ss[i])))227 for j in range(i+1, len(ss)):228 exclusive_c = Clause()229 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, s=ss[i]), negated=True))230 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, s=ss[j]), negated=True))231 axioms.append(exclusive_c)232 axioms.append(disjunctive_c)233 elif ps is not None:234 # NP235 for p in ps:236 disjunctive_c = Clause()237 for i in range(len(ns)):238 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], p=p)))239 for j in range(i+1, len(ns)):240 exclusive_c = Clause()241 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], p=p), negated=True))242 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[j], p=p), negated=True))243 axioms.append(exclusive_c)244 axioms.append(disjunctive_c)245 for n in ns:246 disjunctive_c = Clause()247 for i in range(len(ps)):248 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=n, p=ps[i])))249 for j in range(i+1, len(ps)):250 exclusive_c = Clause()251 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, p=ps[i]), negated=True))252 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, p=ps[j]), negated=True))253 axioms.append(exclusive_c)254 axioms.append(disjunctive_c)255 else:256 # NH257 for h in hs:258 disjunctive_c = Clause()259 for i in range(len(ns)):260 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], h=h)))261 for j in range(i+1, len(ns)):262 exclusive_c = Clause()263 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[i], h=h), negated=True))264 exclusive_c.add_literal(make_literal(make_einstein_prop(n=ns[j], h=h), negated=True))265 axioms.append(exclusive_c)266 axioms.append(disjunctive_c)267 for n in ns:268 disjunctive_c = Clause()269 for i in range(len(hs)):270 disjunctive_c.add_literal(make_literal(make_einstein_prop(n=n, h=hs[i])))271 for j in range(i+1, len(hs)):272 exclusive_c = Clause()273 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, h=hs[i]), negated=True))274 exclusive_c.add_literal(make_literal(make_einstein_prop(n=n, h=hs[j]), negated=True))275 axioms.append(exclusive_c)276 axioms.append(disjunctive_c)277 # Color278 elif cs is not None:279 if bs is not None:280 # CB281 for b in bs:282 disjunctive_c = Clause()283 for i in range(len(cs)):284 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], b=b)))285 for j in range(i+1, len(cs)):286 exclusive_c = Clause()287 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], b=b), negated=True))288 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[j], b=b), negated=True))289 axioms.append(exclusive_c)290 axioms.append(disjunctive_c)291 for c in cs:292 disjunctive_c = Clause()293 for i in range(len(bs)):294 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=c, b=bs[i])))295 for j in range(i+1, len(bs)):296 exclusive_c = Clause()297 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, b=bs[i]), negated=True))298 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, b=bs[j]), negated=True))299 axioms.append(exclusive_c)300 axioms.append(disjunctive_c)301 elif ss is not None:302 # CS303 for s in ss:304 disjunctive_c = Clause()305 for i in range(len(cs)):306 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], s=s)))307 for j in range(i+1, len(cs)):308 exclusive_c = Clause()309 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], s=s), negated=True))310 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[j], s=s), negated=True))311 axioms.append(exclusive_c)312 axioms.append(disjunctive_c)313 for c in cs:314 disjunctive_c = Clause()315 for i in range(len(ss)):316 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=c, s=ss[i])))317 for j in range(i+1, len(ss)):318 exclusive_c = Clause()319 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, s=ss[i]), negated=True))320 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, s=ss[j]), negated=True))321 axioms.append(exclusive_c)322 axioms.append(disjunctive_c)323 elif ps is not None:324 # CP325 for p in ps:326 disjunctive_c = Clause()327 for i in range(len(cs)):328 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], p=p)))329 for j in range(i+1, len(cs)):330 exclusive_c = Clause()331 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], p=p), negated=True))332 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[j], p=p), negated=True))333 axioms.append(exclusive_c)334 axioms.append(disjunctive_c)335 for c in cs:336 disjunctive_c = Clause()337 for i in range(len(ps)):338 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=c, p=ps[i])))339 for j in range(i+1, len(ps)):340 exclusive_c = Clause()341 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, p=ps[i]), negated=True))342 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, p=ps[j]), negated=True))343 axioms.append(exclusive_c)344 axioms.append(disjunctive_c)345 else:346 # CH347 for h in hs:348 disjunctive_c = Clause()349 for i in range(len(cs)):350 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], h=h)))351 for j in range(i+1, len(cs)):352 exclusive_c = Clause()353 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[i], h=h), negated=True))354 exclusive_c.add_literal(make_literal(make_einstein_prop(c=cs[j], h=h), negated=True))355 axioms.append(exclusive_c)356 axioms.append(disjunctive_c)357 for c in cs:358 disjunctive_c = Clause()359 for i in range(len(hs)):360 disjunctive_c.add_literal(make_literal(make_einstein_prop(c=c, h=hs[i])))361 for j in range(i+1, len(hs)):362 exclusive_c = Clause()363 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, h=hs[i]), negated=True))364 exclusive_c.add_literal(make_literal(make_einstein_prop(c=c, h=hs[j]), negated=True))365 axioms.append(exclusive_c)366 axioms.append(disjunctive_c)367 # Beverage368 elif bs is not None:369 if ss is not None:370 # BS371 for s in ss:372 disjunctive_c = Clause()373 for i in range(len(bs)):374 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], s=s)))375 for j in range(i+1, len(bs)):376 exclusive_c = Clause()377 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], s=s), negated=True))378 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[j], s=s), negated=True))379 axioms.append(exclusive_c)380 axioms.append(disjunctive_c)381 for b in bs:382 disjunctive_c = Clause()383 for i in range(len(ss)):384 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=b, s=ss[i])))385 for j in range(i+1, len(ss)):386 exclusive_c = Clause()387 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, s=ss[i]), negated=True))388 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, s=ss[j]), negated=True))389 axioms.append(exclusive_c)390 axioms.append(disjunctive_c)391 elif ps is not None:392 # BP393 for p in ps:394 disjunctive_c = Clause()395 for i in range(len(bs)):396 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], p=p)))397 for j in range(i+1, len(bs)):398 exclusive_c = Clause()399 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], p=p), negated=True))400 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[j], p=p), negated=True))401 axioms.append(exclusive_c)402 axioms.append(disjunctive_c)403 for b in bs:404 disjunctive_c = Clause()405 for i in range(len(ps)):406 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=b, p=ps[i])))407 for j in range(i+1, len(ps)):408 exclusive_c = Clause()409 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, p=ps[i]), negated=True))410 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, p=ps[j]), negated=True))411 axioms.append(exclusive_c)412 axioms.append(disjunctive_c)413 else:414 # BH415 for h in hs:416 disjunctive_c = Clause()417 for i in range(len(bs)):418 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], h=h)))419 for j in range(i+1, len(bs)):420 exclusive_c = Clause()421 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[i], h=h), negated=True))422 exclusive_c.add_literal(make_literal(make_einstein_prop(b=bs[j], h=h), negated=True))423 axioms.append(exclusive_c)424 axioms.append(disjunctive_c)425 for b in bs:426 disjunctive_c = Clause()427 for i in range(len(hs)):428 disjunctive_c.add_literal(make_literal(make_einstein_prop(b=b, h=hs[i])))429 for j in range(i+1, len(hs)):430 exclusive_c = Clause()431 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, h=hs[i]), negated=True))432 exclusive_c.add_literal(make_literal(make_einstein_prop(b=b, h=hs[j]), negated=True))433 axioms.append(exclusive_c)434 axioms.append(disjunctive_c)435 # Smoking436 elif ss is not None:437 if ps is not None:438 # SP439 for p in ps:440 disjunctive_c = Clause()441 for i in range(len(ss)):442 disjunctive_c.add_literal(make_literal(make_einstein_prop(s=ss[i], p=p)))443 for j in range(i+1, len(ss)):444 exclusive_c = Clause()445 exclusive_c.add_literal(make_literal(make_einstein_prop(s=ss[i], p=p), negated=True))446 exclusive_c.add_literal(make_literal(make_einstein_prop(s=ss[j], p=p), negated=True))447 axioms.append(exclusive_c)448 axioms.append(disjunctive_c)449 for s in ss:450 disjunctive_c = Clause()451 for i in range(len(ps)):452 disjunctive_c.add_literal(make_literal(make_einstein_prop(s=s, p=ps[i])))453 for j in range(i+1, len(ps)):454 exclusive_c = Clause()455 exclusive_c.add_literal(make_literal(make_einstein_prop(s=s, p=ps[i]), negated=True))456 exclusive_c.add_literal(make_literal(make_einstein_prop(s=s, p=ps[j]), negated=True))457 axioms.append(exclusive_c)458 axioms.append(disjunctive_c)459 else:460 # SH461 for h in hs:462 disjunctive_c = Clause()463 for i in range(len(ss)):464 disjunctive_c.add_literal(make_literal(make_einstein_prop(s=ss[i], h=h)))465 for j in range(i+1, len(ss)):466 exclusive_c = Clause()467 exclusive_c.add_literal(make_literal(make_einstein_prop(s=ss[i], h=h), negated=True))468 exclusive_c.add_literal(make_literal(make_einstein_prop(s=ss[j], h=h), negated=True))469 axioms.append(exclusive_c)470 axioms.append(disjunctive_c)471 for s in ss:472 disjunctive_c = Clause()473 for i in range(len(hs)):474 disjunctive_c.add_literal(make_literal(make_einstein_prop(s=s, h=hs[i])))475 for j in range(i+1, len(hs)):476 exclusive_c = Clause()477 exclusive_c.add_literal(make_literal(make_einstein_prop(s=s, h=hs[i]), negated=True))478 exclusive_c.add_literal(make_literal(make_einstein_prop(s=s, h=hs[j]), negated=True))479 axioms.append(exclusive_c)480 axioms.append(disjunctive_c)481 # Pet482 else:483 # PH484 for h in hs:485 disjunctive_c = Clause()486 for i in range(len(ps)):487 disjunctive_c.add_literal(make_literal(make_einstein_prop(p=ps[i], h=h)))488 for j in range(i+1, len(ps)):489 exclusive_c = Clause()490 exclusive_c.add_literal(make_literal(make_einstein_prop(p=ps[i], h=h), negated=True))491 exclusive_c.add_literal(make_literal(make_einstein_prop(p=ps[j], h=h), negated=True))492 axioms.append(exclusive_c)493 axioms.append(disjunctive_c)494 for p in ps:495 disjunctive_c = Clause()496 for i in range(len(hs)):497 disjunctive_c.add_literal(make_literal(make_einstein_prop(p=p, h=hs[i])))498 for j in range(i+1, len(hs)):499 exclusive_c = Clause()500 exclusive_c.add_literal(make_literal(make_einstein_prop(p=p, h=hs[i]), negated=True))501 exclusive_c.add_literal(make_literal(make_einstein_prop(p=p, h=hs[j]), negated=True))502 axioms.append(exclusive_c)503 axioms.append(disjunctive_c)504 return axioms505# TODO Implement consistency axioms, which should say that (p_{n,c} and p_{n, p}) \implies p_{c, p}) 506# I.e. (\neg p_{n, c} \vee \neg p_{n, p} \vee p_{c, p}) for all combinations of pairs507def make_consistency_axioms():508 axioms = []509 for i in range(5):510 for j in range(5):511 for k in range(5):512 # NC513 c = Clause()514 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), h=i), negated=True))515 c.add_literal(make_literal(make_einstein_prop(c=Color(k), h=i), negated=True))516 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), c=Color(k))))517 axioms.append(c)518 # NB519 c = Clause()520 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), h=i), negated=True))521 c.add_literal(make_literal(make_einstein_prop(b=Beverage(k), h=i), negated=True))522 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), b=Beverage(k))))523 axioms.append(c)524 525 # NS526 c = Clause()527 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), h=i), negated=True))528 c.add_literal(make_literal(make_einstein_prop(s=Cigar(k), h=i), negated=True))529 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), s=Cigar(k))))530 axioms.append(c)531 # NP532 c = Clause()533 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), h=i), negated=True))534 c.add_literal(make_literal(make_einstein_prop(p=Pet(k), h=i), negated=True))535 c.add_literal(make_literal(make_einstein_prop(n=Nationality(j), p=Pet(k))))536 axioms.append(c)537 # CB538 c = Clause()539 c.add_literal(make_literal(make_einstein_prop(c=Color(j), h=i), negated=True))540 c.add_literal(make_literal(make_einstein_prop(b=Beverage(k), h=i), negated=True))541 c.add_literal(make_literal(make_einstein_prop(c=Color(j), b=Beverage(k))))542 axioms.append(c)543 # CS544 c = Clause()545 c.add_literal(make_literal(make_einstein_prop(c=Color(j), h=i), negated=True))546 c.add_literal(make_literal(make_einstein_prop(s=Cigar(k), h=i), negated=True))547 c.add_literal(make_literal(make_einstein_prop(c=Color(j), s=Cigar(k))))548 axioms.append(c)549 # CP550 c = Clause()551 c.add_literal(make_literal(make_einstein_prop(c=Color(j), h=i), negated=True))552 c.add_literal(make_literal(make_einstein_prop(p=Pet(k), h=i), negated=True))553 c.add_literal(make_literal(make_einstein_prop(c=Color(j), p=Pet(k))))554 axioms.append(c)555 # BS556 c = Clause()557 c.add_literal(make_literal(make_einstein_prop(b=Beverage(j), h=i), negated=True))558 c.add_literal(make_literal(make_einstein_prop(s=Cigar(k), h=i), negated=True))559 c.add_literal(make_literal(make_einstein_prop(b=Beverage(j), s=Cigar(k))))560 axioms.append(c)561 # BP562 c = Clause()563 c.add_literal(make_literal(make_einstein_prop(b=Beverage(j), h=i), negated=True))564 c.add_literal(make_literal(make_einstein_prop(p=Pet(k), h=i), negated=True))565 c.add_literal(make_literal(make_einstein_prop(b=Beverage(j), p=Pet(k))))566 axioms.append(c)567 # SP568 c = Clause()569 c.add_literal(make_literal(make_einstein_prop(s=Cigar(j), h=i), negated=True))570 c.add_literal(make_literal(make_einstein_prop(p=Pet(k), h=i), negated=True))571 c.add_literal(make_literal(make_einstein_prop(s=Cigar(j), p=Pet(k))))572 axioms.append(c)573 return axioms574def make_einstein_axioms():575 axioms = []576 ## Basic exclusion axioms577 ns = [Nationality.brit, Nationality.swede, Nationality.dane, Nationality.norwegian, Nationality.german]578 cs = [Color.red, Color.green, Color.yellow, Color.blue, Color.white]579 bs = [Beverage.tea, Beverage.coffee, Beverage.milk, Beverage.beer, Beverage.water]580 ss = [Cigar.pallmall, Cigar.dunhill, Cigar.blends, Cigar.bluemasters, Cigar.prince]581 ps = [Pet.dogs, Pet.birds, Pet.cats, Pet.horse, Pet.fish]582 hs = [0, 1, 2, 3, 4]583 axioms = axioms + make_exclusion_axioms(ns=ns, cs=cs)584 axioms = axioms + make_exclusion_axioms(ns=ns, bs=bs)585 axioms = axioms + make_exclusion_axioms(ns=ns, ss=ss)586 axioms = axioms + make_exclusion_axioms(ns=ns, ps=ps)587 axioms = axioms + make_exclusion_axioms(ns=ns, hs=hs)588 axioms = axioms + make_exclusion_axioms(cs=cs, bs=bs)589 axioms = axioms + make_exclusion_axioms(cs=cs, ss=ss)590 axioms = axioms + make_exclusion_axioms(cs=cs, ps=ps)591 axioms = axioms + make_exclusion_axioms(cs=cs, hs=hs)592 axioms = axioms + make_exclusion_axioms(bs=bs, ss=ss)593 axioms = axioms + make_exclusion_axioms(bs=bs, ps=ps)594 axioms = axioms + make_exclusion_axioms(bs=bs, hs=hs)595 axioms = axioms + make_exclusion_axioms(ss=ss, ps=ps)596 axioms = axioms + make_exclusion_axioms(ss=ss, hs=hs)597 axioms = axioms + make_exclusion_axioms(ps=ps, hs=hs)598 599 # Consistency axioms600 axioms = axioms + make_consistency_axioms()601 ## Fifteen facts602 # Brit lives in red house603 c1 = Clause()604 c1.add_literal(make_literal(make_einstein_prop(n=Nationality.brit, c=Color.red)))605 axioms.append(c1)606 # Swede keeps dogs607 c2 = Clause()608 c2.add_literal(make_literal(make_einstein_prop(n=Nationality.swede, p=Pet.dogs)))609 axioms.append(c2)610 # Dane drinks tea611 c3 = Clause()612 c3.add_literal(make_literal(make_einstein_prop(n=Nationality.dane, b=Beverage.tea)))613 axioms.append(c3)614 # Green house on the left of white house615 c4 = Clause()616 c4.add_literal(make_literal(make_einstein_prop(c=Color.white, h=1), negated=True))617 axioms.append(c4)618 c4 = Clause()619 c4.add_literal(make_literal(make_einstein_prop(c=Color.white, h=2), negated=True))620 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=1)))621 axioms.append(c4)622 c4 = Clause()623 c4.add_literal(make_literal(make_einstein_prop(c=Color.white, h=3), negated=True))624 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=2)))625 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=1)))626 axioms.append(c4)627 c4 = Clause()628 c4.add_literal(make_literal(make_einstein_prop(c=Color.white, h=4), negated=True))629 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=3)))630 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=2)))631 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=1)))632 axioms.append(c4)633 c4 = Clause()634 c4.add_literal(make_literal(make_einstein_prop(c=Color.white, h=5), negated=True))635 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=4)))636 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=3)))637 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=2)))638 c4.add_literal(make_literal(make_einstein_prop(c=Color.green, h=1)))639 axioms.append(c4)640 # Green house's owner drinks coffee641 c5 = Clause()642 c5.add_literal(make_literal(make_einstein_prop(c=Color.green, b=Beverage.coffee)))643 axioms.append(c5)644 645 # Person who smokes Pall Mall drinks coffee646 c6 = Clause()647 c6.add_literal(make_literal(make_einstein_prop(s=Cigar.pallmall, p=Pet.birds)))648 axioms.append(c6)649 # The owner of the yellow house smokes Dunhill650 c7 = Clause()651 c7.add_literal(make_literal(make_einstein_prop(c=Color.yellow, s=Cigar.dunhill)))652 axioms.append(c7)653 # The man living in the center house drinks milk654 c8 = Clause()655 c8.add_literal(make_literal(make_einstein_prop(b=Beverage.milk, h=2)))656 axioms.append(c8)657 # The Norwegian lives in the first house658 c9 = Clause()659 c9.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=0)))660 axioms.append(c9)661 # The man who smokes Blends lives next to the one who keeps cats662 c10 = Clause()663 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=0), negated=True))664 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=1)))665 axioms.append(c10)666 c10 = Clause()667 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=1)))668 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=0), negated=True))669 axioms.append(c10)670 c10 = Clause()671 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=4), negated=True))672 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=3)))673 axioms.append(c10)674 c10 = Clause()675 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=3)))676 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=4), negated=True))677 axioms.append(c10)678 c10 = Clause()679 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=1), negated=True))680 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=0)))681 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=2)))682 axioms.append(c10)683 c10 = Clause()684 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=2), negated=True))685 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=1)))686 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=3)))687 axioms.append(c10)688 c10 = Clause()689 c10.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=3), negated=True))690 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=2)))691 c10.add_literal(make_literal(make_einstein_prop(p=Pet.cats, h=4)))692 axioms.append(c10)693 # The man who keeps the horse lives next to the man who smokes Dunhill694 c11 = Clause()695 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=0), negated=True))696 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=1)))697 axioms.append(c11)698 c11 = Clause()699 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=1)))700 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=0), negated=True))701 axioms.append(c11)702 c11 = Clause()703 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=4), negated=True))704 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=3)))705 axioms.append(c11)706 c11 = Clause()707 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=3)))708 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=4), negated=True))709 axioms.append(c11)710 c11 = Clause()711 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=1), negated=True))712 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=0)))713 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=2)))714 axioms.append(c11)715 c11 = Clause()716 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=2), negated=True))717 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=1)))718 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=3)))719 axioms.append(c11)720 c11 = Clause()721 c11.add_literal(make_literal(make_einstein_prop(p=Pet.horse, h=3), negated=True))722 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=2)))723 c11.add_literal(make_literal(make_einstein_prop(s=Cigar.dunhill, h=4)))724 axioms.append(c11)725 # The owner who smokes Bluemasters drinks beer726 c12 = Clause()727 c12.add_literal(make_literal(make_einstein_prop(b=Beverage.beer, s=Cigar.bluemasters)))728 axioms.append(c12)729 # The German smokes Prince730 c13 = Clause()731 c13.add_literal(make_literal(make_einstein_prop(n=Nationality.german, s=Cigar.prince)))732 axioms.append(c13)733 # The Norwegian lives next to the blue house734 c14 = Clause()735 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=0), negated=True))736 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=1)))737 axioms.append(c14)738 c14 = Clause()739 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=1)))740 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=0), negated=True))741 axioms.append(c14)742 c14 = Clause()743 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=4), negated=True))744 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=3)))745 axioms.append(c14)746 c14 = Clause()747 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=3)))748 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=4), negated=True))749 axioms.append(c14)750 c14 = Clause()751 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=1), negated=True))752 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=0)))753 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=2)))754 axioms.append(c14)755 c14 = Clause()756 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=2), negated=True))757 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=1)))758 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=3)))759 axioms.append(c14)760 c14 = Clause()761 c14.add_literal(make_literal(make_einstein_prop(n=Nationality.norwegian, h=3), negated=True))762 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=2)))763 c14.add_literal(make_literal(make_einstein_prop(c=Color.blue, h=4)))764 axioms.append(c14)765 # The man who smokes Blends has a neighbor who drinks water766 c15 = Clause()767 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=0), negated=True))768 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=1)))769 axioms.append(c15)770 c15 = Clause()771 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=1)))772 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=0), negated=True))773 axioms.append(c15)774 c15 = Clause()775 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=4), negated=True))776 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=3)))777 axioms.append(c15)778 c15 = Clause()779 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=3)))780 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=4), negated=True))781 axioms.append(c15)782 c15 = Clause()783 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=1), negated=True))784 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=0)))785 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=2)))786 axioms.append(c15)787 c15 = Clause()788 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=2), negated=True))789 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=1)))790 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=3)))791 axioms.append(c15)792 c15 = Clause()793 c15.add_literal(make_literal(make_einstein_prop(s=Cigar.blends, h=3), negated=True))794 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=2)))795 c15.add_literal(make_literal(make_einstein_prop(b=Beverage.water, h=4)))796 axioms.append(c15)797 return axioms798def run():799 assert(make_einstein_prop(n=Nationality.brit, c=Color.red) == 0)800 assert(make_einstein_prop(c=Color.red, b=Beverage.tea) == 125)801 assert(make_einstein_prop(b=Beverage.tea, s=Cigar.pallmall) == 225)802 assert(make_einstein_prop(s=Cigar.pallmall, p=Pet.dogs) == 300)803 assert(make_einstein_prop(p=Pet.dogs, h=0) == 350)804 assert(make_einstein_prop(p=Pet.fish, h=4) == 374)805 axioms = make_einstein_axioms()806 print("p cnf 375 {}".format(len(axioms)))807 for a in axioms:808 print(a)809if __name__ == "__main__":...

Full Screen

Full Screen

08.py

Source:08.py Github

copy

Full Screen

1test = {2 'name': 'Problem 8',3 'points': 2,4 'suites': [5 {6 'cases': [7 {8 'answer': 'a list of restaurants reviewed by the user',9 'choices': [10 'a list of restaurants reviewed by the user',11 'a list of all possible restaurants',12 'a list of ratings for restaurants reviewed by the user'13 ],14 'hidden': False,15 'locked': False,16 'question': r"""17 In the starter code for best_predictor, what does the18 variable reviewed represent?19 """20 },21 {22 'answer': 'a predictor function and its r_squared value',23 'choices': [24 'a predictor function and its r_squared value',25 'a predictor function',26 'an r_squared value',27 'a restaurant'28 ],29 'hidden': False,30 'locked': False,31 'question': r"""32 Given a user, a list of restaurants, and a feature function, what33 does find_predictor from Problem 7 return?34 """35 },36 {37 'answer': 'the predictor with the highest r_squared value',38 'choices': [39 'the predictor with the highest r_squared value',40 'the predictor with the lowest r_squared value',41 'the first predictor in the list',42 'an arbitrary predictor'43 ],44 'hidden': False,45 'locked': False,46 'question': r"""47 After computing a list of [predictor, r_squared] pairs,48 which predictor should we select?49 """50 }51 ],52 'scored': False,53 'type': 'concept'54 },55 {56 'cases': [57 {58 'code': r"""59 >>> user = make_user('Cheapskate', [60 ... make_review('A', 2),61 ... make_review('B', 5),62 ... make_review('C', 2),63 ... make_review('D', 5),64 ... ])65 >>> cluster = [66 ... make_restaurant('A', [5, 2], [], 4, [67 ... make_review('A', 5)68 ... ]),69 ... make_restaurant('B', [3, 2], [], 2, [70 ... make_review('B', 5)71 ... ]),72 ... make_restaurant('C', [-2, 6], [], 4, [73 ... make_review('C', 4)74 ... ]),75 ... make_restaurant('D', [4, 2], [], 2, [76 ... make_review('D', 3),77 ... make_review('D', 4)78 ... ]),79 ... ]80 >>> fns = [restaurant_price, lambda r: mean(restaurant_ratings(r))]81 >>> pred = best_predictor(user, cluster, fns)82 >>> [round(pred(r), 5) for r in cluster] # should be a list of decimals83 [2.0, 5.0, 2.0, 5.0]84 """,85 'hidden': False,86 'locked': False87 },88 {89 'code': r"""90 >>> user = make_user('Cheapskate', [91 ... make_review('A', 2),92 ... make_review('B', 5),93 ... make_review('C', 2),94 ... make_review('D', 5),95 ... ])96 >>> cluster = [97 ... make_restaurant('A', [5, 2], [], 4, [98 ... make_review('A', 5)99 ... ]),100 ... make_restaurant('B', [3, 2], [], 2, [101 ... make_review('B', 5)102 ... ]),103 ... make_restaurant('C', [-2, 6], [], 4, [104 ... make_review('C', 4)105 ... ]),106 ... ]107 >>> fns = [restaurant_price, lambda r: mean(restaurant_ratings(r))]108 >>> pred = best_predictor(user, cluster, fns)109 >>> [round(pred(r), 5) for r in cluster]110 [2.0, 5.0, 2.0]111 """,112 'hidden': False,113 'locked': False114 },115 {116 'code': r"""117 >>> user = make_user('Cheapskate', [118 ... make_review('A', 2),119 ... make_review('B', 5),120 ... make_review('C', 2),121 ... make_review('D', 5),122 ... ])123 >>> cluster = [124 ... make_restaurant('A', [5, 2], [], 4, [125 ... make_review('A', 5)126 ... ]),127 ... make_restaurant('B', [3, 2], [], 2, [128 ... make_review('B', 5)129 ... ]),130 ... make_restaurant('C', [-2, 6], [], 4, [131 ... make_review('C', 4)132 ... ]),133 ... ]134 >>> fns = [lambda r: mean(restaurant_ratings(r)), restaurant_price]135 >>> pred = best_predictor(user, cluster, fns)136 >>> [round(pred(r), 5) for r in cluster] # Make sure you're iterating through feature_fns!137 [2.0, 5.0, 2.0]138 """,139 'hidden': False,140 'locked': False141 },142 {143 'code': r"""144 >>> user = make_user('Cheapskate', [145 ... make_review('A', 2),146 ... make_review('B', 5),147 ... make_review('C', 2),148 ... make_review('D', 5),149 ... ])150 >>> cluster = [151 ... make_restaurant('A', [5, 2], [], 4, [152 ... make_review('A', 5)153 ... ]),154 ... make_restaurant('B', [3, 2], [], 2, [155 ... make_review('B', 5)156 ... ]),157 ... make_restaurant('C', [-2, 6], [], 4, [158 ... make_review('C', 4)159 ... ]),160 ... make_restaurant('E', [1, 2], [], 4, [161 ... make_review('E', 4)162 ... ]),163 ... ]164 >>> fns = [lambda r: mean(restaurant_ratings(r)), restaurant_price]165 >>> pred = best_predictor(user, cluster, fns) # Make sure you're only using user-reviewed restaurants!166 >>> [round(pred(r), 5) for r in cluster]167 [2.0, 5.0, 2.0, 2.0]168 """,169 'hidden': False,170 'locked': False171 }172 ],173 'scored': True,174 'setup': r"""175 >>> import tests.test_functions as test176 >>> from recommend import *177 """,178 'teardown': '',179 'type': 'doctest'180 },181 {182 'cases': [183 {184 'code': r"""185 >>> user = make_user('Cheapskate', [186 ... make_review('A', 2),187 ... make_review('B', 5),188 ... make_review('C', 2),189 ... make_review('D', 5),190 ... ])191 >>> cluster = [192 ... make_restaurant('A', [5, 2], [], 4, [193 ... make_review('A', 5)194 ... ]),195 ... make_restaurant('B', [3, 2], [], 2, [196 ... make_review('B', 5)197 ... ]),198 ... make_restaurant('C', [-2, 6], [], 4, [199 ... make_review('C', 4)200 ... ]),201 ... make_restaurant('D', [4, 2], [], 2, [202 ... make_review('D', 3),203 ... make_review('D', 4)204 ... ]),205 ... ]206 >>> fns = [restaurant_price, lambda r: mean(restaurant_ratings(r))]207 >>> pred = best_predictor(user, cluster, fns)208 >>> # Hint: Price is a perfect predictor of this user's ratings,209 >>> # so the predicted ratings should equal the user's ratings210 >>> [round(pred(r), 5) for r in cluster] # should be a list of decimals211 [2.0, 5.0, 2.0, 5.0]212 """,213 'hidden': False,214 'locked': False215 },216 {217 'code': r"""218 >>> user = make_user('Cheapskate', [219 ... make_review('A', 2),220 ... make_review('B', 5),221 ... make_review('C', 2),222 ... make_review('D', 5),223 ... ])224 >>> cluster = [225 ... make_restaurant('A', [5, 2], [], 4, [226 ... make_review('A', 5)227 ... ]),228 ... make_restaurant('B', [3, 2], [], 2, [229 ... make_review('B', 5)230 ... ]),231 ... make_restaurant('C', [-2, 6], [], 4, [232 ... make_review('C', 4)233 ... ]),234 ... ]235 >>> fns = [restaurant_price, lambda r: mean(restaurant_ratings(r))]236 >>> pred = best_predictor(user, cluster, fns)237 >>> [round(pred(r), 5) for r in cluster]238 [2.0, 5.0, 2.0]239 """,240 'hidden': False,241 'locked': False242 },243 {244 'code': r"""245 >>> user = make_user('Cheapskate', [246 ... make_review('A', 2),247 ... make_review('B', 5),248 ... make_review('C', 2),249 ... make_review('D', 5),250 ... ])251 >>> cluster = [252 ... make_restaurant('A', [5, 2], [], 4, [253 ... make_review('A', 5)254 ... ]),255 ... make_restaurant('B', [3, 2], [], 2, [256 ... make_review('B', 5)257 ... ]),258 ... make_restaurant('C', [-2, 6], [], 4, [259 ... make_review('C', 4)260 ... ]),261 ... ]262 >>> fns = [lambda r: mean(restaurant_ratings(r)), restaurant_price]263 >>> pred = best_predictor(user, cluster, fns)264 >>> [round(pred(r), 5) for r in cluster] # Make sure you're iterating through feature_fns!265 [2.0, 5.0, 2.0]266 """,267 'hidden': False,268 'locked': False269 },270 {271 'code': r"""272 >>> user = make_user('Cheapskate', [273 ... make_review('A', 2),274 ... make_review('B', 5),275 ... make_review('C', 2),276 ... make_review('D', 5),277 ... ])278 >>> cluster = [279 ... make_restaurant('A', [5, 2], [], 4, [280 ... make_review('A', 5)281 ... ]),282 ... make_restaurant('B', [3, 2], [], 2, [283 ... make_review('B', 5)284 ... ]),285 ... make_restaurant('C', [-2, 6], [], 4, [286 ... make_review('C', 4)287 ... ]),288 ... make_restaurant('E', [1, 2], [], 4, [289 ... make_review('E', 4)290 ... ]),291 ... ]292 >>> fns = [lambda r: mean(restaurant_ratings(r)), restaurant_price]293 >>> pred = best_predictor(user, cluster, fns) # Make sure you're only using user-reviewed restaurants!294 >>> [round(pred(r), 5) for r in cluster]295 [2.0, 5.0, 2.0, 2.0]296 """,297 'hidden': False,298 'locked': False299 }300 ],301 'scored': True,302 'setup': r"""303 >>> import tests.test_functions as test304 >>> import recommend305 >>> test.swap_implementations(recommend)306 >>> from recommend import *307 """,308 'teardown': r"""309 >>> test.restore_implementations(recommend)310 """,311 'type': 'doctest'312 }313 ]...

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run lisa automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

... View sample repo... View DocumentationSign up Free
›
_

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

...
Sign up with GoogleSign up with Email
Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful