• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• pydfs-lineup-optimizer-master
  • LICENSE
  • CHANGELOG.md
  • mypy.ini
  • setup.py
  • example.py
  • .travis.yml
  • README.md
  • tests
    • test_optimizer.py
    • test_utils.py
    • test_exporter.py
    • test_rules.py
    • __init__.py
    • utils.py
    • test_late_swap.py
    • players.json
  • requirements.txt
  • pydfs_lineup_optimizer
    • solvers
      • exceptions.py
      • pulp_solver.py
      • constants.py
      • __init__.py
      • base.py
    • rules.py
    • exceptions.py
    • lineup.py
    • lineup_exporter.py
    • version.py
    • player.py
    • constants.py
    • lineup_optimizer.py
    • lineup_importer.py
    • tz.py
    • __init__.py
    • utils.py
    • sites
      • fanball
        • __init__.py
        • settings.py
      • fanduel
        • classic
          • importer.py
          • __init__.py
          • settings.py
        • __init__.py
        • single_game
          • importer.py
          • __init__.py
          • settings.py
      • fantasy_draft
        • importer.py
        • __init__.py
        • settings.py
      • draftkings
        • classic
          • importer.py
          • __init__.py
          • settings.py
        • __init__.py
        • captain_mode
          • importer.py
          • __init__.py
          • settings.py
      • __init__.py
      • sites_registry.py
      • yahoo
        • importer.py
        • __init__.py
        • settings.py
    • settings.py
    • stacks.py
    • lineup_printer.py
  • .gitignore
  • docs
    • Makefile
    • rules.rst
    • export.rst
    • usage.rst
    • performance-and-optimization.rst
    • index.rst
    • conf.py
    • installation.rst
  • tox.ini

from __future__ import absolute_import, division
import unittest
from datetime import datetime, timedelta
from pytz import timezone
from unittest.mock import patch, PropertyMock
from pydfs_lineup_optimizer import get_optimizer
from pydfs_lineup_optimizer.constants import Site, Sport
from pydfs_lineup_optimizer.lineup import Lineup
from pydfs_lineup_optimizer.player import LineupPlayer, GameInfo
from .utils import create_players


class LateSwapTestCase(unittest.TestCase):
    def setUp(self):
        self.future_game_info = GameInfo(home_team='H', away_team='A', game_started=False,
                                         starts_at=datetime.now(timezone('EST')) + timedelta(days=1))
        self.finished_game_info = GameInfo(home_team='H', away_team='A', game_started=False,
                                           starts_at=datetime.now(timezone('EST')) - timedelta(days=1))
        self.lineup_optimizer = get_optimizer(Site.DRAFTKINGS, Sport.BASKETBALL)
        positions = ['PG', 'SG', 'SF', 'PF', 'C', 'PG/SG', 'SF/PF', 'C']
        self.active_players = create_players(positions, game_info=self.future_game_info, salary=5000, fppg=20)
        self.inactive_players = create_players(positions, game_info=self.finished_game_info, salary=4500, fppg=10)
        self.lineup_optimizer.load_players(self.active_players + self.inactive_players)
        self.lineup = Lineup([
            LineupPlayer(self.active_players[0], 'PG'),
            LineupPlayer(self.inactive_players[1], 'SG'),
            LineupPlayer(self.active_players[2], 'SF'),
            LineupPlayer(self.inactive_players[3], 'PF'),
            LineupPlayer(self.active_players[4], 'C'),
            LineupPlayer(self.inactive_players[5], 'G'),
            LineupPlayer(self.active_players[6], 'F'),
            LineupPlayer(self.inactive_players[7], 'UTIL'),
        ])

    def test_late_swap_optimize(self):
        players_in_action = {player: player.lineup_position for player in self.lineup if player.is_game_started}
        lineup = next(self.lineup_optimizer.optimize_lineups([self.lineup]))
        for player in lineup:
            if not player.is_game_started:
                continue
            self.assertIn(player, players_in_action)
            position = players_in_action[player]
            self.assertEqual(position, player.lineup_position)

    def test_late_swap_optimize_with_all_inactive_players(self):
        with patch('pydfs_lineup_optimizer.player.Player.is_game_started', new_callable=PropertyMock) as \
                mock_is_game_started:
            mock_is_game_started.return_value = False
            lineup = next(self.lineup_optimizer.optimize_lineups([self.lineup]))
            for player in lineup:
                self.assertNotIn(player, self.inactive_players)

    def test_late_swap_optimize_with_all_active_players(self):
        with patch('pydfs_lineup_optimizer.player.Player.is_game_started', new_callable=PropertyMock) as \
                mock_is_game_started:
            mock_is_game_started.return_value = True
            lineup = next(self.lineup_optimizer.optimize_lineups([self.lineup]))
            for player, new_lineup_player in zip(self.lineup, lineup):
                self.assertEqual(player, new_lineup_player)