import unittest from trustcafeapiwrapper.utils.get_userprofile_pksk_from_slug import get_userprofile_pksk_from_slug class TestGetUserprofilePkskFromSlug(unittest.TestCase): def test_get_userprofile_pksk_from_slug(self): slug = 'testuser' expected_result = { "pk": "userprofile#testuser", "sk": "userprofile#testuser" } self.assertEqual(get_userprofile_pksk_from_slug(slug), expected_result)