| Index: gm/rebaseline_server/imagepairset_test.py | 
| diff --git a/gm/rebaseline_server/imagepairset_test.py b/gm/rebaseline_server/imagepairset_test.py | 
| index 5e17faa2f13008b73644be22a6e783069684b337..a931e047aa22acffce6b04feea34bc939b51ad52 100755 | 
| --- a/gm/rebaseline_server/imagepairset_test.py | 
| +++ b/gm/rebaseline_server/imagepairset_test.py | 
| @@ -79,9 +79,12 @@ class ImagePairSetTest(unittest.TestCase): | 
| """Assembles some ImagePairs into an ImagePairSet, and validates results. | 
| """ | 
| image_pairs = [ | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_1_AS_DICT), | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_2_AS_DICT), | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_3_AS_DICT), | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_1_AS_DICT), | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_2_AS_DICT), | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_3_AS_DICT), | 
| ] | 
| expected_imageset_dict = { | 
| 'extraColumnHeaders': { | 
| @@ -150,12 +153,14 @@ class ImagePairSetTest(unittest.TestCase): | 
| image_pair_set = imagepairset.ImagePairSet( | 
| diff_base_url=DIFF_BASE_URL) | 
| image_pair_set.add_image_pair( | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_1_AS_DICT)) | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_1_AS_DICT)) | 
| image_pair_set.add_image_pair( | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_2_AS_DICT)) | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_2_AS_DICT)) | 
| with self.assertRaises(Exception): | 
| image_pair_set.add_image_pair( | 
| -          MockImagePair(base_url=BASE_URL_2, | 
| +          MockImagePair(imageA_base_url=BASE_URL_2, imageB_base_url=BASE_URL_2, | 
| dict_to_return=IMAGEPAIR_3_AS_DICT)) | 
|  | 
| def test_missing_column_ids(self): | 
| @@ -164,9 +169,11 @@ class ImagePairSetTest(unittest.TestCase): | 
| image_pair_set = imagepairset.ImagePairSet( | 
| diff_base_url=DIFF_BASE_URL) | 
| image_pair_set.add_image_pair( | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_1_AS_DICT)) | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_1_AS_DICT)) | 
| image_pair_set.add_image_pair( | 
| -        MockImagePair(base_url=BASE_URL_1, dict_to_return=IMAGEPAIR_2_AS_DICT)) | 
| +        MockImagePair(imageA_base_url=BASE_URL_1, imageB_base_url=BASE_URL_1, | 
| +                      dict_to_return=IMAGEPAIR_2_AS_DICT)) | 
| # Call as_dict() with default or reasonable column_ids_in_order. | 
| image_pair_set.as_dict() | 
| image_pair_set.as_dict(column_ids_in_order=['test', 'builder']) | 
| @@ -178,13 +185,14 @@ class ImagePairSetTest(unittest.TestCase): | 
|  | 
| class MockImagePair(object): | 
| """Mock ImagePair object, which will return canned results.""" | 
| -  def __init__(self, base_url, dict_to_return): | 
| +  def __init__(self, imageA_base_url, imageB_base_url, dict_to_return): | 
| """ | 
| Args: | 
| base_url: base_url attribute for this object | 
| dict_to_return: dictionary to return from as_dict() | 
| """ | 
| -    self.base_url = base_url | 
| +    self.imageA_base_url = imageA_base_url | 
| +    self.imageB_base_url = imageB_base_url | 
| self.extra_columns_dict = dict_to_return.get( | 
| imagepair.KEY__IMAGEPAIRS__EXTRACOLUMNS, None) | 
| self._dict_to_return = dict_to_return | 
|  |