Source code for src.FirmChoice

#!/usr/bin/env python

import pandas as pdb
import numpy as np
import copy

class FirmChoice():
    """
        FirmChoice: Contains firm specific data. Implements OOP patterns.
    """
    def __init__(self, family_table, worker_table, curr_choice, curr_esi, c_offer, rtio, prem_df):
        """
            __init__: initializes member variables
        """
        self.__family_table = family_table
        self.__worker_table = worker_table
        self.__curr_choice = curr_choice
        self.__curr_esi = curr_esi
        self.__c_offer = c_offer
        self.__rtio = rtio
        self.__prem_df = prem_df

    """
        Getters
    """
    def get_family_table(self):
        return self.__family_table.copy()

    def get_worker_table(self):
        return self.__worker_table.copy()

    def get_curr_choice(self):
        return self.__curr_choice.copy()

    def get_curr_esi(self):
        return self.__curr_esi.copy()

    def get_c_offer(self):
        return self.__c_offer

    def get_rtio(self):
        return self.__rtio

    def get_prem_df(self):
        return self.__prem_df.copy()

    """
        Mutators
    """
    def set_family_table(self, new_data):
        self.__family_table = new_data

    def set_worker_table(self, new_data):
        self.__worker_table = new_data

    def set_curr_choice(self, new_data):
        self.__curr_choice = new_data

    def set_curr_esi(self, new_data):
        self.__curr_esi = new_data

    def set_c_offer(self, new_data):
        self.__c_offer = new_data

    def set_rtio(self, new_data):
        self.__rtio = new_data

    def set_prem_df(self, new_data):
        self.__prem_df = new_data