def test_sets_correct_data(self):
percentage = Percentage(value=0.5)
data = {
"type": PercentageBucketer.get_type(),
"percentage": percentage.to_dict(),
}
bucketer = PercentageBucketer.from_dict(data)
self.assertEqual(data, bucketer.to_dict())
def test_returns_correct_data(self):
percentage = Percentage(value=0.5)
bucketer = PercentageBucketer(percentage=percentage)
expected = {
"type": PercentageBucketer.get_type(),
"percentage": percentage.to_dict(),
}
self.assertEqual(expected, bucketer.to_dict())
def test_sets_correct_data(self):
percentage = Percentage(value=0.5)
key_whitelist = ["foo"]
data = {
"type": ConsistentHashPercentageBucketer.get_type(),
"key_whitelist": key_whitelist,
"percentage": percentage.to_dict(),
}
bucketer = ConsistentHashPercentageBucketer.from_dict(data)
self.assertEqual(data, bucketer.to_dict())
def test_returns_correct_data(self):
percentage = Percentage(value=0.5)
key_whitelist = ["foo"]
bucketer = ConsistentHashPercentageBucketer(
key_whitelist=key_whitelist, percentage=percentage)
expected = {
"type": ConsistentHashPercentageBucketer.get_type(),
"key_whitelist": key_whitelist,
"percentage": percentage.to_dict(),
}
self.assertEqual(expected, bucketer.to_dict())
def test_returns_correct_values(self):
value = 0.8
percentage = Percentage(value=value)
expected = {"value": value, "type": Percentage.get_type()}
self.assertEqual(expected, percentage.to_dict())
