more unit tests

Author: bmoscon

tests/unit/store/test_version_store_audit.py

@@ -3,11 +3,17 @@
 import pytest
 import pandas as pd
 
-from arctic.store.audit import ArcticTransaction
+from arctic.store.audit import ArcticTransaction, DataChange
 from arctic.store.version_store import VersionedItem, VersionStore
 from arctic.exceptions import ConcurrentModificationException, NoDataFoundException
 
 
+def test_data_change():
+    d = DataChange("a", "b")
+    assert(d.date_range == "a")
+    assert(d.new_data == "b")
+
+
 def test_ArcticTransaction_simple():
     vs = Mock(spec=VersionStore)
     ts1 = pd.DataFrame(index=[1, 2], data={'a':[1.0, 2.0]})

tests/unit/test_multi_index.py

@@ -8,6 +8,7 @@
 import numpy as np
 import pandas as pd
 from tests.util import read_str_as_pandas
+import pytest
 
 
 def get_bitemporal_test_data():
@@ -251,3 +252,9 @@ def test__can_append_row():
     assert len(df) == 5
     assert df.loc[dt('2014-01-05')]['OPEN'] == 9
     assert df.loc[dt('2014-01-05')]['CLOSE'] == 90
+
+
+def test_fancy_group_by_raises():
+    with pytest.raises(ValueError):
+        assert(fancy_group_by(None, method=None))
+