[95] | 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")
|
---|