Insert Delete GetRandom O(1) LeetCode solution in Python

1
class RandomizedSet:
2

3
    def __init__(self):
4
        self.data = set()
5

6
    def insert(self, val: int) -> bool:
7
        if val not in self.data:
8
            self.data.add(val)
9
            return True
10
        return False
11

12
    def remove(self, val: int) -> bool:
13
        if val in self.data:
14
            self.data.remove(val)
15
            return True
16
        return False
17

18
    def getRandom(self) -> int:
19
        return random.choice(list(self.data))