class TestCompanyModel(unittest.TestCase): def setUp(self): self.company = CompanyModel() def test_add_department(self): self.company.add_user('Nisse', 'Donald Duck') assert True def test_get_head_count_for(self): #self.company.add_user('Nisse', 'Donald Duck') self.company.get_headcount_for('Donald Duck') assert True
def step_impl(context): model = getattr(context, "model", None) if not model: context.model = CompanyModel() for row in context.table: context.model.add_user(name=row["name"], department=row["department"]) assert True
def setUp(self): self.company = CompanyModel()