import unittest from trustcafeapiwrapper.utils.make_comment_sk import make_comment_sk class TestMakeCommentSk(unittest.TestCase): def test_make_comment_sk(self): comment_url = '/comment/12345' expected_result = 'comment#12345' self.assertEqual(make_comment_sk(comment_url), expected_result)