Skip to content

Commit

Permalink
Add type hints to tests
Browse files Browse the repository at this point in the history
  • Loading branch information
SkypLabs committed Dec 13, 2024
1 parent 26848a5 commit af18fcf
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/test_wifi_essid.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,11 +20,11 @@ class TestWifiESSID(unittest.TestCase):
Unit tests for the 'WifiESSID' class.
"""

def setUp(self):
def setUp(self) -> None:
self.fake = Faker()
self.fake.add_provider(WifiESSID)

def test_if_string(self):
def test_if_string(self) -> None:
"""
Tests if the values returned by the fake Wi-Fi ESSID generators are
strings.
Expand All @@ -36,7 +36,7 @@ def test_if_string(self):
self.assertTrue(isinstance(self.fake.bbox_default_essid(), str))
self.assertTrue(isinstance(self.fake.wifi_essid(), str))

def test_common_essid(self):
def test_common_essid(self) -> None:
"""
Tests if the 'common_essid()' method returns values from
'COMMON_ESSIDS' as expected.
Expand All @@ -45,7 +45,7 @@ def test_common_essid(self):
for _ in range(10):
self.assertIn(self.fake.common_essid(), COMMON_ESSIDS)

def test_upc_default_essid(self):
def test_upc_default_essid(self) -> None:
"""
Tests if the values returned by the 'upc_default_essid()' method match
the UPC ESSIDs' regex.
Expand All @@ -56,7 +56,7 @@ def test_upc_default_essid(self):
for _ in range(10):
self.assertTrue(re.match(regex, self.fake.upc_default_essid()))

def test_bbox_default_essid(self):
def test_bbox_default_essid(self) -> None:
"""
Tests if the values returned by the 'bbox_default_essid()' method match
the Bbox ESSIDs' regex.
Expand Down

0 comments on commit af18fcf

Please sign in to comment.