1 | import json
|
---|
2 | from pathlib import Path
|
---|
3 | import unittest
|
---|
4 |
|
---|
5 | from pyson.ObjectMapper import ObjectMapper
|
---|
6 |
|
---|
7 | from geniusweb.bidspace.pareto.GenericPareto import GenericPareto
|
---|
8 | from geniusweb.bidspace.pareto.ParetoLinearAdditive import ParetoLinearAdditive
|
---|
9 | from geniusweb.profile.Profile import Profile
|
---|
10 |
|
---|
11 |
|
---|
12 | class ParetoE2Etest (unittest.TestCase):
|
---|
13 | '''
|
---|
14 | Test pareto functionality with some real profiles in an e2e test
|
---|
15 | '''
|
---|
16 | jackson = ObjectMapper()
|
---|
17 |
|
---|
18 | data = \
|
---|
19 | [\
|
---|
20 | [ "test/resources/jobs/jobs1.json", \
|
---|
21 | "test/resources/jobs/jobs2.json", 18 ], \
|
---|
22 | [ "test/resources/7issues/7issues1.json", \
|
---|
23 | "test/resources/7issues/7issues2.json", 557 ]\
|
---|
24 | ]
|
---|
25 |
|
---|
26 | def testAll(self):
|
---|
27 | '''
|
---|
28 | runs all the tests. Python has no standard 'parameterized' mechanism
|
---|
29 | so we have to do it ourselves.
|
---|
30 | '''
|
---|
31 | for [profile1, profile2, expectedNr] in self.data:
|
---|
32 | self.path1 = profile1
|
---|
33 | self.path2 = profile2
|
---|
34 | self.expectedNrPoints = expectedNr
|
---|
35 | with self.subTest("checkGeneric " + self.path1 + "," + self.path2):
|
---|
36 | self.checkGenerciPareto()
|
---|
37 |
|
---|
38 | def checkGenerciPareto(self):
|
---|
39 | file1 = Path(self.path1).read_text("utf-8")
|
---|
40 | profile1 = self.jackson.parse(json.loads(file1), Profile)
|
---|
41 |
|
---|
42 | file2 = Path(self.path2).read_text("utf-8")
|
---|
43 | profile2 = self.jackson.parse(json.loads(file2), Profile)
|
---|
44 |
|
---|
45 | profiles = [profile1, profile2]
|
---|
46 | linaddpareto = ParetoLinearAdditive(profiles)
|
---|
47 | print("linaddpareto result" + str(linaddpareto))
|
---|
48 | self.assertEqual(self.expectedNrPoints, len(linaddpareto.getPoints()))
|
---|
49 |
|
---|
50 | if self.expectedNrPoints < 40:
|
---|
51 | generalpareto = GenericPareto(profiles)
|
---|
52 | self.assertEqual(generalpareto.getPoints(), linaddpareto.getPoints())
|
---|
53 | print("generalpareto result" + str(generalpareto))
|
---|
54 | else:
|
---|
55 | print("Skippedn generalpareto - expected too slow")
|
---|