hikyuu 2.6.8__py3-none-manylinux2014_aarch64.whl

hikyuu/setup.py

@@ -0,0 +1,104 @@
+#!/usr/bin/env python
+#-*- coding:utf-8 -*-
+
+from setuptools import setup, find_packages
+import sys
+import os
+
+if sys.argv[-1] == 'publish':
+    os.system("python setup.py bdist_wheel --python-tag cp35 -p win-amd64")
+    #python setup.py bdist_wheel --python-tag cp35 -p linux_x86_64
+    os.system("twine upload dist/*")
+    sys.exit()
+
+hku_name = "Hikyuu"
+hku_version = "1.0.9"
+hku_author = "fasiondog"
+hku_author_email = "fasiondog@sina.com"
+
+hku_license = "MIT"
+hku_keywords = ("quant", "trade", "System Trading", "backtester", "量化", "程序化交易", "量化交易", "系统交易")
+hku_platforms = "Independant"
+hku_url = "http://hikyuu.org/"
+
+hku_description = "Hikyuu Quant Framework for System Trading Analysis and backtester"
+with open("README.rst", encoding='utf-8') as f:
+    hku_long_description = f.read()
+
+if sys.platform == 'win32':
+    hku_data_files = [('Lib/site-packages/hikyuu', ['hikyuu/hikyuu.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/hikyuu_utils.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/sqlite3.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/importdata.exe']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/boost_date_time-vc141-mt-x64-1_68.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/boost_filesystem-vc141-mt-x64-1_68.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/boost_python35-vc141-mt-x64-1_68.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/boost_serialization-vc141-mt-x64-1_68.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/boost_system-vc141-mt-x64-1_68.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/libmysql.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/hdf5.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/hdf5_hl.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/hdf5_cpp.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/hdf5_hl_cpp.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/szip.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/zlib.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/concrt140.dll']),
+                      ('Lib/site-packages/hikyuu', ['hikyuu/vcruntime140.dll'])]
+else:
+    hku_data_files = []
+
+setup(
+        name = hku_name,
+        version = hku_version,
+        description = hku_description,
+        long_description = hku_long_description,
+        author = hku_author,
+        author_email = hku_author_email,
+
+        license = hku_license,
+        keywords = hku_keywords,
+        platforms = hku_platforms,
+        url = hku_url,
+
+        packages = find_packages(),
+        zip_safe = False,
+        include_package_data=True,
+        package_data = {
+            '': ['*.rst', '*.pyd', '*.ini', '*.sql', '*.properties', '*.xml', 'LICENSE.txt', 'lib*.so.*'],
+        },
+        
+        data_files = hku_data_files,
+        
+        classifiers=[
+            # How mature is this project? Common values are
+            #   3 - Alpha
+            #   4 - Beta
+            #   5 - Production/Stable
+            'Development Status :: 3 - Alpha',
+
+            # Indicate who your project is intended for
+            'Intended Audience :: Developers',
+            'Topic :: Software Development :: Libraries :: Python Modules',
+            'Topic :: Office/Business :: Financial',
+            'Topic :: Office/Business :: Financial :: Investment',
+            'Topic :: Scientific/Engineering :: Mathematics',
+
+            # Pick your license as you wish (should match "license" above)
+             'License :: OSI Approved :: MIT License',
+
+            'Operating System :: Microsoft :: Windows',
+             
+            # Specify the Python versions you support here. In particular, ensure
+            # that you indicate whether you support Python 2, Python 3 or both.
+            'Programming Language :: Python :: 3',
+            'Programming Language :: Python :: 3.5',
+        ],
+
+        install_requires=[
+            'matplotlib>=1.5.0',
+            #'pandas>=0.17.1',
+            #'tushare>=0.8.2',
+            #'pyperclip>=1.5.27',
+            'pyecharts>=0.2.7',
+        ],        
+        )

hikyuu/shell/cmdserver.py

@@ -0,0 +1,126 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2021 hikyuu.org
+#
+# Create on: 2021-03-23
+#    Author: fasiondog
+
+import os, sys, subprocess
+import requests, json
+from hkucmd import HKUShell
+from hikyuu.util.check import hku_catch
+
+g_server_url = "http://127.0.0.1:520/hku"
+g_api_version = "v1"
+g_token = None
+
+
+def get_url(service, api):
+    return "{}/{}/{}/{}".format(g_server_url, service, g_api_version, api)
+
+
+def login():
+    data = json.dumps({'user': 'hku'})
+    url = get_url('login', 'login')
+    r = requests.post(url, data)
+    global g_token
+    g_token = r.json()['hku_token']
+
+
+def logout():
+    url = get_url('login', 'logout')
+    r = requests.post(url, data)
+
+
+def get_token():
+    global g_token
+    if g_token is None:
+        login()
+    return g_token
+
+
+def get_headers():
+    headers = {"hku_token": get_token(), "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"}
+    return headers
+
+
+@hku_catch()
+def start_server():
+    if HKUShell.server is not None:
+        server_status()
+        return
+    target = 'hkuserver.exe' if sys.platform == 'win32' else 'hkuserver'
+    if os.path.exists(target):
+        HKUShell.server = subprocess.Popen([target])
+        return
+
+    platform = 'windows'
+    if sys.platform == 'linux':
+        platform = 'linux'
+    elif sys.platform == 'darwin':
+        platform = 'macosx'
+    target = "./build/release/{}/x64/lib/{}".format(platform, target)
+    if os.path.exists(target):
+        HKUShell.server = subprocess.Popen([target])
+        return
+
+    try:
+        import hikyuu as hku
+        target = "{}/cpp/{}".format(hku.__path__[0], target)
+        if os.path.exists(target):
+            HKUShell.server = subprocess.Popen([target])
+    except:
+        print("Can't found {}".format(target))
+
+
+def stop_server():
+    if HKUShell.server:
+        HKUShell.server.terminate()
+        HKUShell.server.wait()
+        HKUShell.server = None
+        print("server stopped.")
+
+
+def server_status():
+    if HKUShell.server is None:
+        print("server stopped.")
+        return
+    url = get_url('assist', 'status')
+    headers = get_headers()
+    r = requests.get(url, headers=headers)
+    print(r.json())
+
+
+def set_server_logger_level(logger, level):
+    url = get_url('assist', 'log_level')
+    headers = get_headers()
+    data = {'logger': logger, 'level': int(level)} if logger is not None else {'level': int(level)}
+    r = requests.post(url, data, headers=headers)
+    print(r.json())
+
+
+def server(self, args):
+    """
+    start: 启动服务，无参数时，默认为start
+    stop: 停止服务
+    status: 查看当前服务器运行状态
+    set_logger_level: 设置 logger 级别 
+        logger(str): （可选）logger 名称
+        level(int): 打印级别
+    """
+    if args == "":
+        start_server()
+    x = args.split()
+    if not x or x[0] == 'status':
+        server_status()
+    elif x[0] == 'stop':
+        stop_server()
+    elif x[0] == 'set_logger_level':
+        if len(x) == 2:
+            set_server_logger_level(None, x[1])
+        else:
+            set_server_logger_level(x[1], x[2])

hikyuu/shell/hkucmd.py

@@ -0,0 +1,110 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2021 hikyuu.org
+#
+# Create on: 2021-03-09
+#    Author: fasiondog
+
+import cmd
+import functools
+
+
+def shell_cmd(func):
+    @functools.wraps(func)
+    def wrapcmd(*args, **kargs):
+        try:
+            self = args[0]
+            self.__class__.lineno += 1
+            self.__class__.prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % self.__class__.lineno
+            result = func(*args, **kargs)
+            print("")
+            return result
+        except Exception as e:
+            print(e)
+
+    return wrapcmd
+
+
+class HKUShell(cmd.Cmd):
+    intro = """\033[1;32;40m
+    ooooo   ooooo ooooo oooo    oooo oooooo   oooo ooooo     ooo ooooo     ooo 
+    `888'   `888' `888' `888   .8P'   `888.   .8'  `888'     `8' `888'     `8' 
+     888     888   888   888  d8'      `888. .8'    888       8   888       8  
+     888ooooo888   888   88888[         `888.8'     888       8   888       8  
+     888     888   888   888`88b.        `888'      888       8   888       8  
+     888     888   888   888  `88b.       888       `88.    .8'   `88.    .8'  
+    o888o   o888o o888o o888o  o888o     o888o        `YbodP'       `YbodP'    
+    \033[0m
+    \033[33m\033[1m  Welcome to the hikyuu shell. Type help or ? to list commands.\033[0m
+
+    """
+    lineno = 1
+    prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % lineno
+    file = None
+    server = None
+
+    def __del__(self):
+        if HKUShell.server is not None:
+            HKUShell.server.terminate()
+            HKUShell.server.wait()
+            HKUShell.server = None
+
+    @classmethod
+    def add_command(cls, func):
+        """
+        def func(self, args):
+            do something, don't return
+        """
+        @functools.wraps(func)
+        def wrapcmd(*args, **kargs):
+            try:
+                self = args[0]
+                self.__class__.lineno += 1
+                self.__class__.prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % self.__class__.lineno
+                result = func(*args, **kargs)
+                print("")
+                return result
+            except Exception as e:
+                print(e)
+                print("")
+
+        if func.__name__ == "do_":
+            raise Exception("Invalid func name!")
+        name = func.__name__ if len(func.__name__) > 3 and func.__name__[:3] == "do_" else "do_%s" % func.__name__
+        setattr(cls, name, wrapcmd)
+
+    @shell_cmd
+    def do_hello(self, arg):
+        """You are welcome"""
+        print(arg)
+
+    def do_quit(self, arg):
+        'Stop recording, and exit:  BYE'
+        print('Thank you for using hikyuu.\n')
+        return True
+
+    # ----- record and playback -----
+    def do_record(self, arg):
+        'Save future commands to filename:  RECORD rose.cmd'
+        self.file = open(arg, 'w')
+
+    def do_playback(self, arg):
+        'Playback commands from a file:  PLAYBACK rose.cmd'
+        self.close()
+        with open(arg) as f:
+            self.cmdqueue.extend(f.read().splitlines())
+
+    def precmd(self, line):
+        line = line.lower()
+        if self.file and 'playback' not in line:
+            print(line, file=self.file)
+        return line
+
+    def close(self):
+        if self.file:
+            self.file.close()
+            self.file = None

hikyuu/shell/hkushell.py

@@ -0,0 +1,20 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2021 hikyuu.org
+#
+# Create on: 2021-03-09
+#    Author: fasiondog
+
+from hkucmd import *
+from cmdserver import server
+
+HKUShell.add_command(server)
+
+if __name__ == "__main__":
+    import colorama
+    colorama.init(autoreset=True)
+    HKUShell().cmdloop()

hikyuu/strategy/__init__.py

@@ -0,0 +1,2 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-

hikyuu/strategy/strategy_demo1.py

@@ -0,0 +1,53 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+
+#
+# 警告：Hikyuu 为量化研究工具，本身不包含程序化交易接口。此部分仅为策略调度运行时示例，
+#      供自行实现程序化交易时参考，请自行负责程序化交易可能造成的损失。
+#
+
+from hikyuu import *
+
+
+def on_change(stg: Strategy, stk: Stock, spot: SpotRecord):
+    print("[on_change]:", stk.market_code, stk.name, spot.close, spot.bid[0], spot.ask[0])
+
+
+def on_spot(stg: Strategy, rev_time: Datetime):
+    print("[on_received_spot] rev_time:", rev_time)
+
+
+def my_func1(stg: Strategy):
+    print("[my_func1]", str(stg.now()))
+
+
+def my_func2(stg: Strategy):
+    print("[my_func2] calculate:", stg.now())
+    for s in sm:
+        print(s)
+
+
+# 注意：
+#   1.每一个Strategy 只能作为独立进程执行，即 python xxx.py 的方式执行！
+#   2.请开启 HikyuuTdx 行情采集，否则接收不到数据
+# Strategy 方式运行示例
+if __name__ == '__main__':
+    # 创建策略运行时，必须指定 stock 和 ktype 列表
+    # strategy 只会加载指定的 stock, ktype 的数据，行情接收也只会更新这些数据
+    # 如需使用交易日历，请记得同时指定 sh000001
+    s = Strategy(['sh600000', 'sz000001'],  [Query.MIN, Query.DAY])
+
+    # 当前自动延迟10秒/20秒后执行，忽略节假日限制
+    s.run_daily_at(my_func1, Datetime.now() - Datetime.today() + Seconds(10), False)
+    s.run_daily_at(my_func1, Datetime.now() - Datetime.today() + Seconds(20), False)
+
+    # 收到指定 stock 的行情更新
+    s.on_change(on_change)
+
+    # 收到行情更新
+    s.on_received_spot(on_spot)
+
+    # 每隔 1 分钟循环一次 (ignore_market 忽略开闭市时间限制, 否则仅在开盘期间执行)
+    s.run_daily(my_func2, Minutes(1))  # , ignore_market=True)
+    s.start()

hikyuu/strategy/strategy_demo2.py

@@ -0,0 +1,47 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+
+import easytrader
+from hikyuu import *
+
+#
+# 警告：Hikyuu 为量化研究工具，本身不包含程序化交易接口。此部分仅为策略调度运行时示例，
+#      供自行实现程序化交易时参考，请自行负责程序化交易可能造成的损失。
+#
+
+# 创建 easytrade 订单代理（示例仅支持华泰）可自行参照 EasyTraderOrderBroker 修改
+# buy|sell 中已屏蔽实际通过easytrade下单，防止调试误操作
+user = easytrader.use('ht_client')
+user.connect(r'D:\htwt\xiadan.exe')
+easy_ob = EasyTraderOrderBroker(user)
+broker = crtOB(easy_ob)
+
+# 获取交易系统策略
+# 目前 run_in_strategy 只支持非延迟交易，即收盘时立刻买入/卖出
+sys = get_part("default.sys.趋势双均线")
+sys.set_param("buy_delay", False)
+sys.set_param("sell_delay", False)
+
+
+# 执行策略主体
+def my_func(stg: Strategy):
+    # 这里示例使用的是 TC_Zero() 零成本算法，但实际建议使用接近实际的成本算法
+    # 因为 sys 不依赖于成交单中的实际成本，而是使用成本算法预算需要买入的数量
+    run_in_strategy(sys, sm['sz000001'], Query(Datetime(20240101)), broker, TC_Zero())
+
+
+# 注意：
+#   1.每一个Strategy 只能作为独立进程执行，即 python xxx.py 的方式执行！
+#   2.请开启 HikyuuTdx 行情采集，否则接收不到数据
+# Strategy 方式运行示例
+if __name__ == '__main__':
+    # 创建策略运行时，必须指定 stock 和 ktype 列表
+    # strategy 只会加载指定的 stock, ktype 的数据，行情接收也只会更新这些数据
+    # 如需使用交易日历，请记得同时指定 sh000001
+    s = Strategy(['sh000001', 'sz000001'],  [Query.DAY])
+
+    # 每交易日 14点55分 执行
+    s.run_daily_at(my_func, TimeDelta(0, 14, 55))
+    s.start()
+
+    # 上述，也可以参见 strategy_demo3.py 中使用 crt_sys_strategy 快捷创建 strategy

hikyuu/strategy/strategy_demo3.py

@@ -0,0 +1,24 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+
+#
+# 警告：Hikyuu 为量化研究工具，本身不包含程序化交易接口。此部分仅为策略调度运行时示例，
+#      供自行实现程序化交易时参考，请自行负责程序化交易可能造成的损失。
+#
+
+if __name__ == '__main__':
+    import easytrader
+    from hikyuu import *
+
+    # 创建 easytrade 订单代理（示例仅支持华泰）可自行参照 EasyTraderOrderBroker 修改
+    user = easytrader.use('ht_client')
+    user.connect(r'D:\htwt\xiadan.exe')
+    easy_ob = EasyTraderOrderBroker(user)
+    broker = crtOB(easy_ob)
+
+    sys = get_part("default.sys.趋势双均线")
+
+    # 直接使用 sys 创建 strategy 示例，如果为日线，则自动每日 14点55分 执行
+    # 如果 query 为日线以下（分钟线、5分钟线）则自动按对应间隔（分钟、5分钟）循环执行
+    stg = crt_sys_strategy(sys, "sz000001", Query(Datetime(20240101), ktype=Query.DAY), broker, TC_Zero(), "demo3")
+    stg.start()

hikyuu/test/AllocateFunds.py

@@ -0,0 +1,25 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+#
+# Create on: 2022-02-20
+#    Author: fasiondog
+
+import unittest
+from test_init import *
+
+
+class SystemWeightTest(unittest.TestCase):
+    def test_weight(self):
+        x = SystemWeight()
+        self.assertEqual(x.weight, 1.0)
+
+        try:
+            x.weight = 1.2
+        except:
+            self.assertEqual(x.weight, 1.0)
+
+        #print(x)
+
+
+def suite():
+    return unittest.TestLoader().loadTestsFromTestCase(SystemWeightTest)

hikyuu/test/Condition.py

@@ -0,0 +1,95 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# gb18030
+
+# ===============================================================================
+# 作者：fasiondog
+# 历史：1）20130311, Added by fasiondog
+# ===============================================================================
+
+import unittest
+
+from test_init import *
+from hikyuu import *
+
+
+class ConditionPython(ConditionBase):
+    def __init__(self):
+        super(ConditionPython, self).__init__("ConditionPython")
+        self.set_param("n", 10)
+        self._m_flag = False
+
+    def is_valid(self, datetime):
+        return self._m_flag
+
+    def _reset(self):
+        if self._m_flag:
+            self._m_flag = False
+        else:
+            self._m_flag = True
+
+    def _clone(self):
+        p = ConditionPython()
+        p._m_flag = self._m_flag
+        return p
+
+
+class ConditionTest(unittest.TestCase):
+    def test_ConditionBase(self):
+        p = ConditionPython()
+        self.assertEqual(p.name, "ConditionPython")
+        self.assertEqual(p.get_param("n"), 10)
+        p.set_param("n", 20)
+        self.assertEqual(p.get_param("n"), 20)
+        self.assertEqual(p.is_valid(Datetime(200101010000)), False)
+        p.reset()
+        self.assertEqual(p.is_valid(Datetime(200101010000)), True)
+
+        p_clone = p.clone()
+        self.assertEqual(p_clone.name, "ConditionPython")
+        self.assertEqual(p_clone.get_param("n"), 20)
+        self.assertEqual(p_clone.is_valid(Datetime(200101010000)), True)
+
+        p.set_param("n", 1)
+        p_clone.set_param("n", 3)
+        self.assertEqual(p.get_param("n"), 1)
+        self.assertEqual(p_clone.get_param("n"), 3)
+
+
+def testCondition(self):
+    self._add_valid(Datetime(201108120000))
+    self._add_valid(Datetime(201108250000))
+
+
+class TestCrtCN(unittest.TestCase):
+    def test_crtCN(self):
+        p = crtCN(testCondition, params={'n': 10}, name='ConditionPython')
+        self.assertEqual(p.get_param("n"), 10)
+        p.set_param("n", 20)
+        self.assertEqual(p.get_param("n"), 20)
+
+        k = sm['sh000001'].get_kdata(Query(-100))
+        self.assertEqual(k.empty(), False)
+        p.sg = SignalBase()  # cn设置交易对象时，必须已经指定了信号指示器
+        p.to = k  # cn在设置交易对象时才会调用_caculate函数
+        self.assertEqual(p.is_valid(Datetime(201108120000)), True)
+        self.assertEqual(p.is_valid(Datetime(201108250000)), True)
+        self.assertEqual(p.is_valid(Datetime(200101030000)), False)
+
+        p_clone = p.clone()
+        self.assertEqual(p_clone.name, "ConditionPython")
+        self.assertEqual(p_clone.get_param("n"), 20)
+        self.assertEqual(p_clone.is_valid(Datetime(201108120000)), True)
+
+        p.set_param("n", 1)
+        p_clone.set_param("n", 3)
+        self.assertEqual(p.get_param("n"), 1)
+        self.assertEqual(p_clone.get_param("n"), 3)
+
+
+def suite():
+    return unittest.TestLoader().loadTestsFromTestCase(ConditionTest)
+
+
+def suiteTestCrtCN():
+    return unittest.TestLoader().loadTestsFromTestCase(TestCrtCN)

hikyuu/test/Datetime.py

@@ -0,0 +1,61 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+
+# ===============================================================================
+# 作者：fasiondog
+# 历史：1）20120927, Added by fasiondog
+# ===============================================================================
+
+import unittest
+
+from test_init import *
+
+
+class DatetimeTest(unittest.TestCase):
+    def test_Datetime(self):
+        d = Datetime(201209272301)
+        self.assertEqual(d.year, 2012)
+        self.assertEqual(d.month, 9)
+        self.assertEqual(d.day, 27)
+        self.assertEqual(d.hour, 23)
+        self.assertEqual(d.minute, 1)
+        self.assertEqual(d.number, 201209272301)
+        self.assertEqual(Datetime(140001010000), Datetime.min())
+        self.assertEqual(Datetime(999912310000), Datetime.max())
+
+        self.assertEqual(d, Datetime(201209272301))
+        self.assertNotEqual(d, Datetime(200101010000))
+        self.assertGreaterEqual(d, Datetime(201209272301))
+        self.assertTrue(d >= Datetime(201209272259))
+        self.assertTrue(d <= Datetime(201209272301))
+        self.assertTrue(d <= Datetime(201209272302))
+        self.assertTrue(not (d > Datetime(201209272301)))
+        self.assertTrue(d > Datetime(201209272259))
+        self.assertTrue(not (d < Datetime(201209272301)))
+        self.assertTrue(d < Datetime(201209272302))
+
+        d = Datetime(200101010159)
+        self.assertEqual(str(d), "2001-01-01 01:59:00")
+        self.assertEqual(d, Datetime("2001-Jan-01 01:59:00"))
+
+        self.assertEqual(Datetime(), constant.null_datetime)
+
+    def test_pickle(self):
+        if not constant.pickle_support:
+            return
+
+        import pickle as pl
+        a = Datetime(201001010000)
+        filename = sm.tmpdir() + "/Datetime.plk"
+        fh = open(filename, 'wb')
+        pl.dump(a, fh)
+        fh.close()
+        fh = open(filename, 'rb')
+        b = pl.load(fh)
+        fh.close()
+        self.assertEqual(a, b)
+
+
+def suite():
+    return unittest.TestLoader().loadTestsFromTestCase(DatetimeTest)