def test_cond_select_values(self, category_data): dcM = DataCategory('A', category_data['attributeListMiss'], category_data['rowListUnicodeMiss']) atL = dcM.getAttributeList() for ii, at in enumerate(atL[1:]): assert len( dcM.selectIndices(category_data['testRowUnicodeMiss'][ii], at)) == dcM.getRowCount() # print("Window %r" % [tt for tt in window(atL)]) for atW in window(atL, size=1): assert len( dcM.selectValueListWhere( atW, category_data['testRowUnicodeMiss'][-1], category_data['attributeListMiss'] [-1])) == dcM.getRowCount() for atW in window(atL, size=2): assert len( dcM.selectValueListWhere( atW, category_data['testRowUnicodeMiss'][-1], category_data['attributeListMiss'] [-1])) == dcM.getRowCount() for atW in window(atL, size=3): assert len( dcM.selectValueListWhere( atW, category_data['testRowUnicodeMiss'][-1], category_data['attributeListMiss'] [-1])) == dcM.getRowCount() for atW in window(atL, size=4): assert len( dcM.selectValueListWhere( atW, category_data['testRowUnicodeMiss'][-1], category_data['attributeListMiss'] [-1])) == dcM.getRowCount()
def testCondSelectValues(self): """Test case - value selections - """ try: dcM = DataCategory("A", self.__attributeListMiss, self.__rowListUnicodeMiss) # self.__testRowUnicodeMiss = [u'someData', 100222, None, '?', '.', u'abcdĆćĈĉĊċČčĎďĐđĒēĔĕĖėĘęĚěĜĝĞğĠġĢģĤĥĦħĨxyz', 234.2345] # self.__attributeListMiss atL = dcM.getAttributeList() for ii, at in enumerate(atL[1:]): self.assertEqual( len(dcM.selectIndices(self.__testRowUnicodeMiss[ii], at)), dcM.getRowCount()) # logger.debug("Window %r", [tt for tt in window(atL)]) for atW in window(atL, size=1): self.assertEqual( len( dcM.selectValueListWhere( atW, self.__testRowUnicodeMiss[-1], self.__attributeListMiss[-1])), dcM.getRowCount()) for atW in window(atL, size=2): self.assertEqual( len( dcM.selectValueListWhere( atW, self.__testRowUnicodeMiss[-1], self.__attributeListMiss[-1])), dcM.getRowCount()) for atW in window(atL, size=3): self.assertEqual( len( dcM.selectValueListWhere( atW, self.__testRowUnicodeMiss[-1], self.__attributeListMiss[-1])), dcM.getRowCount()) for atW in window(atL, size=4): self.assertEqual( len( dcM.selectValueListWhere( atW, self.__testRowUnicodeMiss[-1], self.__attributeListMiss[-1])), dcM.getRowCount()) except Exception as e: logger.exception("Failing with %s", str(e)) self.fail()