def test_base_methods_unicode(self, category_data): name = 'A' dcbA = DataCategoryBase(name, category_data['attributeList'], category_data['rowListUnicode']) assert name == dcbA.getName() assert category_data['attributeList'] == dcbA.getAttributeList() assert category_data['rowListUnicode'] == dcbA.getRowList() assert len(category_data['rowListUnicode']) == dcbA.getRowCount() ii = 0 dcbA.setMapping('DATA') for row in dcbA: ii += 1 assert category_data['testRowUnicode'] == row[1:] assert ii == dcbA.getRowCount() dcbA.setMapping('ATTRIBUTE') ii = 0 na = len(dcbA.getAttributeList()) for row in dcbA: ii += 1 # logger.info("ii %d row %r " % (ii, row)) assert ii == row['colOrd'] assert category_data['testRowUnicode'][na - 2] == row['colD'] # assert ii == dcbA.getRowCount() dcbA.setMapping('ITEM') ii = 0 for row in dcbA: ii += 1 assert ii == row['_' + name + '.' + 'colOrd'] assert category_data['testRowUnicode'][3] == row['_' + name + '.' + 'colD'] assert ii == dcbA.getRowCount()
def test_base_methods_ascii(self, category_data): name = 'A' dcbA = DataCategoryBase(name, category_data['attributeList'], category_data['rowListAsciiA']) assert name == dcbA.getName() assert category_data['attributeList'] == dcbA.getAttributeList() assert category_data['rowListAsciiA'] == dcbA.getRowList() assert len(category_data['rowListAsciiA']) == dcbA.getRowCount() ii = 0 dcbA.setMapping('DATA') for row in dcbA: ii += 1 assert category_data['testRowAsciiA'] == row[1:] assert ii == dcbA.getRowCount() dcbA.setMapping('ATTRIBUTE') ii = 0 na = len(dcbA.getAttributeList()) for row in dcbA: ii += 1 assert ii == row['colOrd'] assert category_data['testRowAsciiA'][na - 2] == row['colD'] assert ii == dcbA.getRowCount() dcbA.setMapping('ITEM') ii = 0 for row in dcbA: ii += 1 assert ii == row['_' + name + '.' + 'colOrd'] assert ii == dcbA.getRowCount()
def testBaseMethodsAscii(self): """Test case - base class methods""" try: name = "A" dcbA = DataCategoryBase(name, self.__attributeList, self.__rowListAsciiA) self.assertEqual(name, dcbA.getName()) self.assertEqual(self.__attributeList, dcbA.getAttributeList()) self.assertEqual(self.__rowListAsciiA, dcbA.getRowList()) self.assertEqual(len(self.__rowListAsciiA), dcbA.getRowCount()) ii = 0 dcbA.setMapping("DATA") for row in dcbA: ii += 1 self.assertEqual(self.__testRowAsciiA, row[1:]) self.assertEqual(ii, dcbA.getRowCount()) dcbA.setMapping("ATTRIBUTE") ii = 0 na = len(dcbA.getAttributeList()) for row in dcbA: ii += 1 # logger.info("ii %d row %r " % (ii, row)) self.assertEqual(ii, row["colOrd"]) self.assertEqual(self.__testRowAsciiA[na - 2], row["colD"]) # self.assertEqual(ii, dcbA.getRowCount()) dcbA.setMapping("ITEM") ii = 0 for row in dcbA: ii += 1 self.assertEqual(ii, row["_" + name + "." + "colOrd"]) self.assertEqual(ii, dcbA.getRowCount()) except Exception as e: logger.exception("Failing with %s", str(e)) self.fail()