def check_set_tier_separate(test_tier_array):
    test_health_plan = create_plan(test_tier_array[0], separate=True)

    calc._set_metal_tier(test_health_plan)
    integrated_table = ContinuanceTable.find(session, test_tier_array[1] + "Integrated")
    medical_table = ContinuanceTable.find(session, test_tier_array[1] + "Medical")
    assert (
        calc._integrated_table == integrated_table
    ), "Integrated Table is not correct for tier {0}, actual {1} expected {2}".format(
        test_tier_array[0], calc._integrated_table, integrated_table
    )
    assert (
        calc._medical_table == medical_table
    ), "Medical Table is not correct for tier {0}, actual {1} expected {2}".format(
        test_tier_array[0], calc._integrated_table, medical_table
    )
    def _set_metal_tier(self, healthplan):
        if healthplan.tier == Tier.na or healthplan.tier == None:
            tier_name = 'Platinum'
        else:
            tier_name = healthplan.tier.name.capitalize()

        self.__separate = (healthplan.medical_deductible != None)

        self._integrated_table = ContinuanceTable.find(self.__session, tier_name + 'Integrated')
        if self.__separate:
            self._medical_table = ContinuanceTable.find(self.__session, tier_name + 'Medical')
            self._drug_table = ContinuanceTable.find(self.__session, tier_name + 'Drug')
        else:
            self._medical_table = self._integrated_table
            self._drug_table = self._integrated_table
        self._determine_total_cost()
def check_set_tier_separate(test_tier_array):
    test_health_plan = create_plan(test_tier_array[0], separate=True)

    calc._set_metal_tier(test_health_plan)
    integrated_table = ContinuanceTable.find(session,
                                             test_tier_array[1] + 'Integrated')
    medical_table = ContinuanceTable.find(session,
                                          test_tier_array[1] + 'Medical')
    assert calc._integrated_table == integrated_table, \
        "Integrated Table is not correct for tier {0}, actual {1} expected {2}".format(test_tier_array[0],
                                                                                       calc._integrated_table,
                                                                                       integrated_table)
    assert calc._medical_table == medical_table, \
        "Medical Table is not correct for tier {0}, actual {1} expected {2}".format(test_tier_array[0],
                                                                                   calc._integrated_table,
                                                                                   medical_table)
 def test_find(self):
     test_table1 = ContinuanceTable(name='Test1', membership=1000, avg_cost=1000)
     test_table2 = ContinuanceTable(name='Test2', membership=1500, avg_cost=100)
     self.session.add(test_table1)
     self.session.add(test_table2)
     self.session.commit()
     found_table = ContinuanceTable.find(self.session, "Test1")
     assert found_table.id == test_table1.id, 'Table was not found'
    def _set_metal_tier(self, healthplan):
        if healthplan.tier == Tier.na or healthplan.tier == None:
            tier_name = 'Platinum'
        else:
            tier_name = healthplan.tier.name.capitalize()

        self.__separate = (healthplan.medical_deductible != None)

        self._integrated_table = ContinuanceTable.find(
            self.__session, tier_name + 'Integrated')
        if self.__separate:
            self._medical_table = ContinuanceTable.find(
                self.__session, tier_name + 'Medical')
            self._drug_table = ContinuanceTable.find(self.__session,
                                                     tier_name + 'Drug')
        else:
            self._medical_table = self._integrated_table
            self._drug_table = self._integrated_table
        self._determine_total_cost()