from copy import deepcopy

import unittest

from booking$\_$system import get$\_$most$\_$popular$\_$countries as get$\_$mpcs

class TestGetMostPopularCountries$($unittest$.$TestCase$)$:

$"""$Test the function get$\_$most$\_$popular$\_$countries."""

def setUp$($self$)$:

self.example $=\{$

$\text{'}137\text{'}$: $\{$

'identifier': $\text{'}137\text{'},$

'headline': 'Beautiful $4-$bedroom home on lakeside',

'listed': $\text{'}2024-07-17\text{'},$

'features': $[\text{'}4$ bedroom', $\text{'}3$ bath', 'Pool'$],$

'location': $(\text{'}$New York', 'USA'$),$

'description': 'This is a beautiful $4-$bedroom, $3-$bath detached'

$\text{'}$ house

with an outdoor pool.$\text{'}\},$

$\text{'}001\text{'}$: $\{$

'identifier': $\text{'}001\text{'},$

'headline': 'House',

'listed': None,

'features': $[\text{'}$clean$\text{'}],$

'location': $(\text{'}$Hamilton$\text{'},$ 'Canada'$),$

'description': None$\},$

$\text{'}05\text{'}$: $\{$

'identifier': $\text{'}05\text{'},$

'headline': 'Stunning lakeside cottage$\text{'},$

'listed': $\text{'}2021-02-08\text{'},$

'features': $[],$

'location': $(\text{'}$Oxford$\text{'},\text{'}$UK$\text{'}),$

'description': 'This is lakeside!$\text{'}\}$

$\}$

def test$\_$handout$\_$example$($self$)$:

$"""$Test get$\_$most$\_$popular$\_$countries with the handout example."""

book$\_$sys$\_$copy $=$ deepcopy$($self$.$example$)$

expected $=[\text{'}$Canada$\text{'},\text{'}$UK$\text{'},$ 'USA'$]$

actual $=$ get$\_$mpcs$($self$.$example$)$

msg $=$ message$($book$\_$sys$\_$copy, expected, actual$)$

self.assertEqual$($actual$,$ expected, msg$)$

# TODO: add a complete test suite here

def message$($test$\_$case: dict, expected: list, actual: object$)->$ str:

$"""$Return an error message saying the function call

get$\_$most$\_$popular$\_$countries$($test$\_$case$)$ resulted in the value

actual, when the correct value is expected.

$"""$

return $(\text{'}$When we called get$\_$most$\_$popular$\_$countries$(\text{'}+$ str$($test$\_$case$)$

$+\text{'})$ we expected $\text{'}+$ str$($expected$)$

$+\text{'},$ but got $\text{'}+$ str$($actual$))$

if $\_\_$name$\_\_==\text{'}\_\_$main$\_\_\text{'}$:

unittest.main$($exit$=$False$)$