mamadyonline

"""Test data base module"""
import unittest

from data_engineering.database.fake_players import generate_fake_player
from data_engineering.database.players import Player
from data_engineering.database.postgres_db import DataBase


class TestDB(unittest.TestCase):
    def setUp(self) -> None:

mamadyonline

from faker import Faker
import random

COUNTRIES = ['Argentina', 'Portugal', 'France', 'Brazil', 
             'England', 'Belgium', 'Italy', 'Spain', 'Germany',
             'Ivory coast', 'Guinea', 'Senegal', 'Ghana', 
             'Nigeria', 'Cameroun', 'Egypt', 'Morocco', 'Algeria',
             'USA', 'Croatia', 'Sweden', 'Columbia', 'Netherlands', 
             'Danemark', 'Ireland', 'Switzerland']

mamadyonline

"""Class defining some queries for the database via SQLAlchemy"""
from typing import List

from data_engineering.database.base import Base, Session, data_base
from data_engineering.database.players import Player


class DataBase:

    def __init__(self):

mamadyonline

"""Create players class and define its characteristics"""

from data_engineering.database.base import Base
from sqlalchemy import String, Integer, Column, Boolean


class Player(Base):
    __tablename__ = 'players'

    id = Column(Integer, primary_key=True)

mamadyonline

"""This code creates:

* a SQLAlchemy Engine that will interact with our dockerized PostgreSQL database,
* a SQLAlchemy ORM session factory bound to this engine,
* a base class for our classes definitions.
"""

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

mamadyonline

from decimal import Decimal

# We can construct decimal numbers with integers, floats and strings
# Using integers  and floats return the exact representation of the
# as stored by Python, on 64 bits
a = Decimal(0.1) # Decimal('0.1000000000000000055511151231257827021181583404541015625')
b = Decimal('0.1') # Decimal('0.1')

# It also works with NaN and Infinity
decimal_nan = Decimal('NaN')

mamadyonline

mamadyonline

mamadyonline

import numpy as np

class PSO(object):
  """
    Class implementing PSO algorithm.
  """
  def __init__(self, func, init_pos, n_particles):
    """
      Initialize the key variables.