gh-20 - finished tests and fixed issues in gaffer.py

gchq · Jun 28, 2017 · cdf1714 · cdf1714
1 parent 6e49c1b
commit cdf1714
Show file tree

Hide file tree

Showing 2 changed files with 156 additions and 157 deletions.
diff --git a/python-shell/src/gafferpy/gaffer.py b/python-shell/src/gafferpy/gaffer.py
@@ -486,8 +486,8 @@ def to_json(self):
 
         element_generator = {'class': self.generator_class_name}
         if self.element_generator_fields is not None:
-            for field in self.element_generator_fields:
-                element_generator[field.key] = field.value
+            for key, value in self.element_generator_fields.items():
+                element_generator[key] = value
         operation['elementGenerator'] = element_generator
         return operation
 
@@ -518,8 +518,8 @@ def to_json(self):
 
         element_generator = {'class': self.generator_class_name}
         if self.element_generator_fields is not None:
-            for field in self.element_generator_fields:
-                element_generator[field.key] = field.value
+            for key, value in self.element_generator_fields.items():
+                element_generator[key] = value
         operation['elementGenerator'] = element_generator
         return operation