First version

tests/create_custom.py

@@ -0,0 +1,13 @@
+from keepit import keepit
+
+ORIGIN = None
+
+
+@keepit('myfunction')
+def custom_function():
+    return ORIGIN
+
+
+if __name__ == '__main__':
+    ORIGIN = 'main'
+    custom_function()

tests/test.tsv

@@ -0,0 +1,3 @@
+	int	str	bool	other
+0	0	hola	True	
+1	1	adiós	False	5.0

tests/test_main.py

@@ -0,0 +1,98 @@
+import sys
+import os
+import subprocess
+from unittest import TestCase
+
+from keepit import keepit, diff, hash_df
+from keepit.backends import Pickle
+
+import pandas as pd
+
+# df = pd.DataFrame([[0, 'hola', True, None], [1, 'adiós', False, 5]], columns=['int', 'str', 'bool', 'other'])
+
+this_directory = os.path.abspath(os.path.dirname(__file__))
+
+
+count = 0
+
+@keepit()
+def prueba(first, name='something', value=29):
+    global count
+    count += 1
+    return pd.DataFrame([[count, first, name, value], ], columns=['time', 'first', 'name', 'value'])
+
+
+@keepit()
+def prueba_df():
+    return pd.read_csv(os.path.join(this_directory, 'test.tsv'), sep='\t')
+
+
+@keepit()
+def prueba_df_argument(df):
+    return count
+
+
+class TestMain(TestCase):
+
+
+    def setUpClass():
+        back = Pickle()
+        back.erase_all()
+        assert not back.find()
+        assert (not os.path.exists(back.res_folder)) or (not os.listdir(back.res_folder))
+
+    def tearDownClass():
+        back = Pickle()
+        back.erase_all()
+
+    def test_basic(self):
+
+        try:
+            prueba()
+        except TypeError:
+            pass
+
+        p1 = prueba('hello')
+        print(p1)
+        p2 = prueba('hello')
+        print(p2)
+        print(diff(p1, p2))
+        assert p1.equals(p2)  # Columns and rows are equal
+
+        p3 = prueba('other')
+        print(p3)
+        assert not p1.equals(p3)
+
+        p4 = prueba('hello', name='different')
+        print(p4)
+        assert not p1.equals(p4)
+
+    def test_list_and_drop(self):
+        assert len(prueba.list()) > 0
+        prueba.drop_all()
+        assert len(prueba.list()) == 0
+
+    def test_df(self):
+        df1 = prueba_df()
+        df2 = prueba_df()
+
+        assert df1.equals(df2)
+        assert hash_df(df1) == hash_df(df2)
+        prueba_df.drop_all()
+
+    def test_different_program(self):
+        '''A value saved by a different interpreter should be reusable.'''
+        from create_custom import custom_function
+        custom_function.drop_all()
+        subprocess.check_call([sys.executable, os.path.join(this_directory, 'create_custom.py')])
+        # create_custom returns different results when run as a script.
+        # We make sure we are using the "stored" value, and not the result from the imported function
+        assert custom_function() == 'main'
+
+    def test_df_arg(self):
+
+        df1 = pd.read_csv(os.path.join(this_directory, 'test.tsv'), sep='\t')
+        res1 = prueba_df_argument(df1)
+        df2 = pd.read_csv(os.path.join(this_directory, 'test.tsv'), sep='\t')
+        res2 = prueba_df_argument(df2)
+        assert res1 == res2