def test_03_token_prefix(self): prefix = get_token_prefix("totp") self.assertTrue(prefix == "TOTP", prefix) prefix = get_token_prefix("X_Y_Z", "does not exist") self.assertTrue(prefix == "does not exist", prefix) prefix = get_token_prefix() self.assertTrue(prefix.get("totp") == "TOTP", prefix) self.assertTrue(prefix.get("hotp") == "OATH", prefix)