1 files changed, 75 insertions, 0 deletions
diff --git a/weave/tests/test_common_spec.py b/weave/tests/test_common_spec.py
new file mode 100755
index 000000000..b51c39892
--- /dev/null
+++ b/weave/tests/test_common_spec.py
@@ -0,0 +1,75 @@
+import unittest
+
+from scipy_distutils.misc_util import add_grandparent_to_path, restore_path
+
+add_grandparent_to_path(__name__)
+import inline_tools
+restore_path()
+    
+class test_file_specification(unittest.TestCase):    
+    def check_py_to_file(self):
+        import tempfile
+        file_name = tempfile.mktemp()        
+        file = open(file_name,'w')
+        code = """
+               fprintf(file,"hello bob");
+               """
+        inline_tools.inline(code,['file']) 
+        file.close()
+        file = open(file_name,'r')
+        assert(file.read() == "hello bob")
+    def check_file_to_py(self):
+        import tempfile
+        file_name = tempfile.mktemp()        
+        # not sure I like Py::String as default -- might move to std::sting
+        # or just plain char*
+        code = """
+               char* _file_name = PyString_AsString( file_name.ptr() );
+               FILE* file = fopen(_file_name,"w");
+               Py::Object file_obj(file_to_py(file,_file_name,"w"));
+               return_val = Py::new_reference_to(file_obj);
+               """
+        file = inline_tools.inline(code,['file_name'])
+        file.write("hello fred")        
+        file.close()
+        file = open(file_name,'r')
+        assert(file.read() == "hello fred")
+
+class test_instance_specification(unittest.TestCase):    
+    pass
+    
+class test_callable_specification(unittest.TestCase):        
+    def check_call_function(self):
+        import string
+        func = string.find
+        search_str = "hello world hello"
+        sub_str = "world"
+        # * Not sure about ref counts on search_str and sub_str.
+        # * Is the Py::String necessary? (it works anyways...)
+        code = """
+               Py::Tuple args(2);
+               args[0] = Py::String(search_str);
+               args[1] = Py::String(sub_str);
+               PyObject* result = PyObject_CallObject(func,args.ptr());
+               return_val = Py::new_reference_to(Py::Int(result));
+               """
+        actual = inline_tools.inline(code,['func','search_str','sub_str'])
+        desired = func(search_str,sub_str)        
+        assert(desired == actual)
+
+def test_suite():
+    suites = []   
+    suites.append( unittest.makeSuite(test_file_specification,'check_'))
+    suites.append( unittest.makeSuite(test_instance_specification,'check_'))
+    suites.append( unittest.makeSuite(test_callable_specification,'check_'))
+    total_suite = unittest.TestSuite(suites)
+    return total_suite
+
+def test():
+    all_tests = test_suite()
+    runner = unittest.TextTestRunner()
+    runner.run(all_tests)
+    return runner
+
+if __name__ == "__main__":
+    test()