def test_successful_with_flop_turn_river_cards(self):
data = {
'runs': 10,
'user_hand': ['', ''],
'additional_players': 2,
'additional_hands': [['', ''], ['H14', '']],
'flop_cards': ['D3', 'D4', 'D5'],
'turn_card': 'D14',
'river_card': 'C14'
}
serializer = SimulationSerializer(data=data)
valid = serializer.is_valid()
print(serializer.errors)
simulation = serializer.save()
self.assertTrue(valid)
self.assertEqual(simulation.user_hand, ['', ''])
self.assertEqual(simulation.additional_hands, [['', ''], ['H14', '']])
simulation.run_simulation()
self.assertTrue('H14' in simulation.all_players[2].get_hand())
for player in simulation.all_players:
self.assertEqual(len(player.hand), 2)
def test_serializer_creates_simulation_with_starter_hands(self):
data = {
'runs': 1000,
'user_hand': ['D2', 'D4'],
'additional_players': 2,
'additional_hands': [['D14', 'C14'], ['H5', 'H6']]
}
serializer = SimulationSerializer(data=data)
if serializer.is_valid():
simulation = serializer.save()
results = simulation.run_simulation()
self.assertTrue(serializer.is_valid())
self.assertEqual(simulation.runs, 1000)
self.assertTrue(['D2', 'D4'] == simulation.user.starting_hand)
self.assertTrue(
['D14', 'C14'] == simulation.all_players[1].starting_hand
or ['D14', 'C14'] == simulation.all_players[2].starting_hand)
self.assertTrue(
['H5', 'H6'] == simulation.all_players[1].starting_hand
or ['H5', 'H6'] == simulation.all_players[2].starting_hand)
self.assertEqual(len(simulation.all_players), 3)
def post(self, request, *args, **kwargs):
print(request.data)
serializer = SimulationSerializer(data=request.data)
if serializer.is_valid():
simulation = serializer.save()
simulation.run_simulation()
return Response(serializer.data)
else:
print(serializer.errors)
return Response(serializer.errors)
def test_serializer_creates_simulation_without_starter_hands(self):
data = {
'runs': 1000,
'user_hand': [],
'additional_players': 2,
'additional_hands': []
}
serializer = SimulationSerializer(data=data)
if serializer.is_valid():
simulation = serializer.save()
results = simulation.run_simulation()
self.assertTrue(serializer.is_valid())
self.assertEqual(simulation.runs, 1000)
self.assertEqual(len(simulation.all_players), 3)
def test_successful_with_single_cards(self):
data = {
'runs': 10,
'user_hand': ['D14', ''],
'additional_players': 2,
'additional_hands': [['', 'C14'], ['H14', '']]
}
serializer = SimulationSerializer(data=data)
valid = serializer.is_valid()
simulation = serializer.save()
self.assertTrue(valid)
self.assertEqual(simulation.user_hand, ['D14', ''])
self.assertEqual(simulation.additional_hands,
[['', 'C14'], ['H14', '']])
simulation.run_simulation()
self.assertTrue('D14' in simulation.user.get_hand())
self.assertTrue('C14' in simulation.all_players[1].get_hand())
self.assertTrue('H14' in simulation.all_players[2].get_hand())