GH83_DawidSkeneTest.java

package test.java.integration.algorithms.gal;

import com.datascience.core.base.AssignedLabel;
import com.datascience.core.base.LObject;
import com.datascience.core.base.Worker;
import com.datascience.core.nominal.CategoryValue;
import com.datascience.core.nominal.NominalProject;
import com.datascience.core.nominal.decision.LabelProbabilityDistributionCostCalculators;
import com.datascience.core.nominal.decision.WorkerEstimator;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import static org.junit.Assert.assertFalse;

/**
 * @Author: konrad
 */
public class GH83_DawidSkeneTest extends AbstractBase {

	@Override
	protected List<String> getCategories(){
		return Arrays.asList(new String[]{"yes", "no", "blank"});
	}

	@Override
	public void runTestScenario(NominalProject project) {

		List<CategoryValue> priors = new ArrayList<CategoryValue>();
		for (String s : getCategories()) {
			priors.add(new CategoryValue(s, 1. / 3.));
		}

		project.initializeCategories(getCategories(), priors, null);

		Worker w1 = new Worker("A2G061ZJVQXGE");
		Worker w2 = new Worker("A224TK7J4KA1LV");
		Worker w3 = new Worker("A2ZUENR4ZLC3MN");
		project.getData().addAssign(new AssignedLabel<String>(w1, new LObject<String>("21K74J79KQW6KZ5ZI8OFKD455U4HJ6"), "blank"));
		project.getData().addAssign(new AssignedLabel<String>(w1, new LObject<String>("2FWPLSP75KLMPN1Y0O8QMQVJZMCZTJ"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("283PHIT3HVWAEB01E1AQA1727RZCKR"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2S3YHVI44OS23CC3P15Z0BAPDTU4Y4"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2NCXFCD45XCECEVNZ7BIV3TGXMSYWJ"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("295QQ7Q670519BUI3FTTYC1XUY7WC6"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("255NVRH1KHO9X796JKEVHLKWFFDZQ5"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("21Z1Q6SVQ8H8RD8PIZWOJVCI99NN4U"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2JGU98JHSTLBMCIANX0IWBJER94EQE"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2Z3KH1Q6SVQ80ZQH28XLDOBVK0S2LC"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2P3Z6R70G5R9R4QEF9574SSTVG2P9O"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("25UCF0CP5KXPC95DUBPGCPM7VGWWX9"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2AOYTWX4H3H282M8LN7IEIJRJNW4ZT"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2J5JQ2172Z9998O6WZ3VMBU1B1FZR6"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2SIHFL42J1LYOZT2ELD0QGFKPVS2NE"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2W36VCPWZ9RNWN0J7FJ70N3D1YKPA3"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2F8S1CSTMOFXA4AZUQSE6DNXRXCROW"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2TRKSSZVXX2QNWCPW4FYONIW95IN5Z"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2L13NAB6BFMFYFGTU7STRFDGMA2HU2"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2B3RGNTBJ3MMWONA1LE6AKHBH439VT"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("24ZMVHVKCJ01KBK2GXY9SQW69QD758"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2V99Y8TNKIMZBDNFRVJ2WM0POD59W3"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2FK4E6AUBXHWB8PAXN0VJZWDZ3NX1B"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2WAVGULF395SFX5BDTJ3VYNJJG5PBY"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2DMZMAZHHRRLYHIJZ1MRP1KHWR4CLU"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("27E618N46UXFV4M09Q5TVDSN128RPD"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2J9CJK63ZVE30IIF6W1QHJL5ZL59Y2"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2DS6X3YOCCF0VGNF93KIVIIX6YURQ4"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("23ZGYQ4J3NABP2XHRTFY6IT1115PC5"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2PFMB3Q39Z0BPLBJI7VE3STVZ26UJK"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("283FQ0ONNVRHKBZJLS7RJ76N3RBULM"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2F8S1CSTMOFXA4AZUQSE6DNXRXFORW"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("21UJ011K274JQ02L8KS8V46U5X5BD9"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2QAKMT6YTWX40UZX1PVDH9GIE0C0VB"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2C18EBDPGFL6E37RBNINLWIO3JLH2M"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2Y6VR14IXKO240KCQQNAEX3YWU59AV"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2P7JWKUDD8XMUU8YLDRBEUTOEBXMB2"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2525UFJ51Y7QXFHAA4J1KSTMWXQ9CX"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2NCXFCD45XCECEVNZ7BIV3TGXMQWYF"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("27ZM8JIA0RYN20KIVYD0DBMJ1C8507"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2I63W5XH0JPPB9KFASGL0P75S3FEKP"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("21RRJ85OZIZE0DS510FOY5EWXFB9XW"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2LAMEZZ2MIKM60C31NP3GTWW50TN6F"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("241KM05BMJTU0PEXS7G9ZA7SDEFEJQ"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2UZDM25L8I9NINOCQDXOM4QWFBYHRY"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2J1E1M7PKK9LANOJBO30CLZXX3YZSU"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2LWGDHDM25L8105U8K8E76OEC8MEO6"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2XANTL0XULRG6KTEF0DD55FPF5X2OT"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2FC98JHSTLB34RX6VN9OJJEJZWORFD"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2D2UDD8XMB3QM0HVNKLTW6T4MDIEPN"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2C73TVOK5UFJOSG22SFZNYQS9UI743"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("25XXRDS4IC1EH45SVTD19A2I3GUWZ6"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2PG1THV8ER9YRK5FU0QSU5KFPAT2PA"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2C79746OQ1SN9HPLILR59QKCV9RN39"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2YDKCJ011K27NAP4W4N698N4ECQ9BQ"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2XGQ4J3NAB6BYDXA0CPI11TJNV6ERF"), "yes"));//,F
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("23K5L8I9NZW605H10SVQ47T8ER6UKJ"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2ZCUU00OQKKAN90NK6TUA680MEQWWG"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2P1HSTLB3L0FUARD0PAERREVMNHUI8"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2581SNQQ7Q67JWJLWQER92TQKJN9TD"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2LISPW1INMHGHHMEF11BEBFMNXHH4I"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2SIHFL42J1LYOZT2ELD0QGFKPVRN2Y"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2ZRNZW6HEZ6OXV8RJ7Z6HGUMAJRZPE"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("214T6YTWX4H30T76GR09OI6IR91X2S"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2T5HMVHVKCJ0KS2XJIA7HKQWEJ1646"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2ALDHJHP4BDD37MTK5JXHX3GBNY4XW"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2IDP9746OQ1S6H822KY0D1QKK5I2ME"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2WIOHOVR14IX3FKGLG8EKWA65LO76O"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2GYKPEIB9BAW4J4S14WV5UD3UBHX08"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2WQ06UFBNFSVZL3AFNWS46NG9XLH3X"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2L13NAB6BFMFYFGTU7STRFDGMA2UHF"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2ALDHJHP4BDD37MTK5JXHX3GBN04XY"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2F4NCWYB49F9BJAC9FSJWNSTDZYOSK"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2BXBNFSVGULFM0NN8KEG9FS3VGE7LX"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("20JLY58B727MJ9YAWV41EYFS3XXU9S"), "no"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2FKAOO2GMMEG7FYL4QD6QAA20W877N"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("20JLY58B727MJ9YAWV41EYFS3XU9U4"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2A79RA7S5WM1P0GN0TY0RMHLI0JZUA"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w2, new LObject<String>("2AOYTWX4H3H282M8LN7IEIJRJNY4ZV"), "yes"));
		project.getData().addAssign(new AssignedLabel<String>(w3, new LObject<String>("2K5AB6BFMFFOHP0OD7AFLGESKIHJW8"), "yes"));

		ensureComputed(project);
		WorkerEstimator we = new WorkerEstimator(LabelProbabilityDistributionCostCalculators.get("ExpectedCost"));
		for (Double d : we.getCosts(project).values()) {
			assertFalse(Double.isNaN(d));
		}
	}
}