From da35aece6826d2e92ede30b9bef044516bf45815 Mon Sep 17 00:00:00 2001
From: Mark <mfrimston@gmail.com>
Date: Fri, 27 May 2011 10:03:26 +0100
Subject: [PATCH 1/2] Made changes for 2.6

---
 src/mrf/codegen.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/mrf/codegen.py b/src/mrf/codegen.py
index f7c9585..c836097 100644
--- a/src/mrf/codegen.py
+++ b/src/mrf/codegen.py
@@ -1039,6 +1039,7 @@ def __invert__(self):
 		
 if __name__ == "__main__":	
 	import unittest
+	import sys
 	
 	class MockFunctionBuilder(object):
 	
@@ -1066,6 +1067,10 @@ def check_created(self,testcase,vals):
 	
 	
 	class TestStatementBuilder(unittest.TestCase):
+
+		if sys.version_info < (2,7):
+			def assertIs(self, a,b):
+				self.assertTrue(a is b)
 	
 		def setUp(self):
 			self.o = MockFunctionBuilder()
@@ -1118,8 +1123,8 @@ def test_is_not_statement(self):
 			self.o.check_combined(self,[self.t])
 			
 		def test_and(self):
-			self.s.and_(5.7)
-			self.assertEquals("foobar and 5.7", repr(self.s))
+			self.s.and_(5.0)
+			self.assertEquals("foobar and 5.0", repr(self.s))
 			
 		def test_and_statement(self):
 			self.s.and_(self.t)
@@ -1364,8 +1369,8 @@ def test_isub_statement(self):
 
 		def test_imul(self):
 			ss = self.s
-			self.s *= 8.8
-			self.assertEquals("foobar *= 8.8", repr(ss))
+			self.s *= 8.0
+			self.assertEquals("foobar *= 8.0", repr(ss))
 			
 		def test_imul_statement(self):
 			ss = self.s
@@ -1491,6 +1496,10 @@ class MockStatement(object):
 
 	class TestFunctionBuilderInterface(unittest.TestCase):
 	
+		if sys.version_info < (2,7):
+			def assertIs(self,a,b):
+				self.assertTrue(a is b)
+	
 		def setUp(self):
 			self.s = MockStatement()
 			self.o = MockFunctionBuilder(self.s)
@@ -1510,5 +1519,6 @@ def test_get_attribute_special(self):
 				s = getattr(self.i, n+"_")
 				self.assertIs(self.s,s)
 				self.o.check_created(self,[n])
+
 			
 	unittest.main()

From 3f64639572d6c84551d3412adb9a79eb17a5cd7d Mon Sep 17 00:00:00 2001
From: Mark <mfrimston@gmail.com>
Date: Fri, 27 May 2011 17:42:48 +0100
Subject: [PATCH 2/2] added some tests including doctest

---
 src/mrf/codegen.py | 62 +++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 58 insertions(+), 4 deletions(-)

diff --git a/src/mrf/codegen.py b/src/mrf/codegen.py
index c836097..7d92616 100644
--- a/src/mrf/codegen.py
+++ b/src/mrf/codegen.py
@@ -1,6 +1,19 @@
 # TODO: unit tests
 # TODO: documentation
 
+"""	
+Facilitates the dynamic generation of functions using an intuitive syntax.
+
+	>>> with def_("myfunction",globals=globals(),locals=locals()) as f:
+	... 	with f.for_( f.i, f.range(3) ):
+	... 		f.print_("Hi")
+	...
+	>>> myfunction()
+	Hi
+	Hi
+	Hi
+
+"""
 
 def def_(name,args=[],kargs={},globals={},locals={}):
 	"""	
@@ -252,9 +265,10 @@ def end_block(self):
 		
 	def __exit__(self,extype,exvalue,traceback):
 		"""	
-		Invoked when context is exited
+		Invoked when context is exited. The function is automatically built.
 		"""
 		self.state = FunctionBuilder.STATE_AFTER
+		self.build()
 		
 	def build(self):
 		"""	
@@ -1040,6 +1054,7 @@ def __invert__(self):
 if __name__ == "__main__":	
 	import unittest
 	import sys
+	import doctest
 	
 	class MockFunctionBuilder(object):
 	
@@ -1491,7 +1506,8 @@ def test_invert(self):
 
 
 	class MockStatement(object):
-		pass
+		def __repr__(self):
+			return "MOCK"
 
 
 	class TestFunctionBuilderInterface(unittest.TestCase):
@@ -1519,6 +1535,44 @@ def test_get_attribute_special(self):
 				s = getattr(self.i, n+"_")
 				self.assertIs(self.s,s)
 				self.o.check_created(self,[n])
+				
+		def test_set_attribute(self):
+			self.i.foo = "bar"	
+			self.o.check_created(self,["foo = 'bar'"])
 
-			
-	unittest.main()
+		def test_set_existing_attribute(self):
+			self.i._owner = "bar"
+			self.o.check_created(self,[])
+			
+		def test_brackets(self):
+			s = self.i.b_(77)
+			self.assertIs(self.s, s)
+			self.o.check_created(self,["( 77 )"])
+			
+		def test_brackets_statement(self):
+			t = StatementBuilder(self.o,"foo")
+			s = self.i.b_(t)
+			self.assertIs(self.s, s)
+			self.o.check_created(self,["( foo )"])
+			self.o.check_combined(self,[t])
+				
+		def test_lit(self):
+			s = self.i.lit_("hi")
+			self.assertIs(self.s, s)
+			self.o.check_created(self,["'hi'"])
+				
+		def test_not(self):
+			s = self.i.not_(20)
+			self.assertIs(self.s, s)
+			self.o.check_created(self,["not 20"])
+				
+		def test_not_statement(self):
+			t = StatementBuilder(self.o,"foo")
+			s = self.i.not_(t)
+			self.assertIs(self.s,s)
+			self.o.check_created(self,["not foo"])
+			self.o.check_combined(self,[t])
+				
+
+	doctest.testmod()							
+	unittest.main()