Merge pull request pandas-dev#94 from manahl/coverage

new unit tests

Author: bmoscon

tests/unit/date/test_mktz.py

@@ -3,7 +3,7 @@
 from pytest import raises
 
 from arctic.date import mktz, TimezoneError
-from arctic.date._mktz import DEFAULT_TIME_ZONE_NAME
+from arctic.date._mktz import DEFAULT_TIME_ZONE_NAME, tzfile, TIME_ZONE_DATA_SOURCE
 
 
 def test_mktz():
@@ -37,3 +37,9 @@ def test_mktz_fails_if_invalid_timezone():
         file_exists.return_value = False
         with raises(TimezoneError):
             mktz('junk')
+
+
+def test_tzfile_raises():
+    t = tzfile(TIME_ZONE_DATA_SOURCE + DEFAULT_TIME_ZONE_NAME)
+    with raises(ValueError):
+        t._find_ttinfo(None) 

tests/unit/date/test_util.py

@@ -86,3 +86,23 @@ def test_to_dt_dt_tz():
 
 def test_to_dt_dt_tz_default():
     assert to_dt(dt(1970, 1, 1, tzinfo=mktz('UTC')), mktz('Europe/London')) == dt(1970, 1, 1, tzinfo=mktz('UTC'))
+
+
+def test_daterange_raises():
+    with pytest.raises(ValueError):
+        assert(DateRange(dt(2013, 1, 1), dt(2000, 1, 1)))
+
+
+def test_daterange_eq():
+    dr = DateRange(dt(2013, 1, 1))
+    assert((dr == None) == False)
+    assert(dr == dr)
+
+
+def test_daterange_lt():
+    dr = DateRange(dt(2013, 1, 1))
+    dr2 = DateRange(dt(2001, 1, 1))
+
+    assert(dr2 < dr)
+    dr.start = None
+    assert((dr2 < dr) == False)

tests/unit/store/test_version_item.py

@@ -24,3 +24,12 @@ def test_versioned_item_str_handles_none():
                          metadata=None)
 
     assert str(item)
+
+
+def test_versioned_item_metadata_dict():
+    item = VersionedItem(symbol="test",
+                         library="test_lib",
+                         data=None,
+                         version=1.2,
+                         metadata=None)
+    assert(item.metadata_dict() == {'symbol': 'test', 'library': 'test_lib', 'version': 1.2})

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(sentinel, sentinel)
+    assert(d.date_range == sentinel)
+    assert(d.new_data == sentinel)
+
+
 def test_ArcticTransaction_simple():
     vs = Mock(spec=VersionStore)
     ts1 = pd.DataFrame(index=[1, 2], data={'a':[1.0, 2.0]})

tests/unit/test_compression.py

@@ -1,5 +1,5 @@
 from mock import patch, Mock, sentinel, call
-from arctic._compression import compress, compress_array, decompress, decompress_array
+from arctic._compression import compress, compress_array, decompress, decompress_array, enable_parallel_lz4
 import lz4
 
 
@@ -68,3 +68,13 @@ def test_compress_array_no_parallel():
         assert decompress_array(compress_array(a)) == a
         assert patch_lz4.compress.call_args_list == [call(x) for x in a]
         assert patch_lz4.decompress.call_args_list == [call(compress(x)) for x in a]
+
+
+def test_enable_parallel_lz4():
+    enable_parallel_lz4(True)
+    from arctic._compression import ENABLE_PARALLEL
+    assert(ENABLE_PARALLEL == True)
+    enable_parallel_lz4(False)
+    from arctic._compression import ENABLE_PARALLEL
+    assert(ENABLE_PARALLEL == False)
+

tests/unit/test_fixtures.py

@@ -0,0 +1,20 @@
+import arctic.fixtures.arctic as fix
+from mock import Mock
+from arctic.arctic import Arctic
+
+
+def test_overlay_library_name():
+    assert(fix.overlay_library_name() == 'test.OVERLAY')
+
+
+def test_overlay_library():
+    a = Mock(Arctic, autospec=True)
+    fix.overlay_library(a, "test")
+    a.initialize_library.assert_called_with("test_RAW", "VersionStore", segment='year')
+
+
+def test_tickstore_lib():
+    a = Mock(Arctic, autospec=True)
+    fix.tickstore_lib(a, "test")
+    a.initialize_library.assert_called_with('test', 'TickStoreV3')
+    a.get_library.assert_called_with('test') 

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))
+

tests/unit/test_util.py

@@ -0,0 +1,15 @@
+from arctic._util import are_equals, enable_sharding
+from mock import MagicMock, ANY
+from arctic.arctic import Arctic
+
+
+def test_are_equals_not_df():
+    assert(are_equals(1.0, 2.0) is False)
+    assert(are_equals([1, 2, 3], [1,2,3]))
+    assert(are_equals("Hello", "World") is False)
+
+
+def test_enable_sharding_hashed():
+    m = MagicMock(Arctic, autospec=True)
+    enable_sharding(m, "test", hashed=True)
+    m._conn.admin.command.assert_called_with('shardCollection', ANY, key={'symbol': 'hashed'})

tests/unit/tickstore/test_toplevel.py

@@ -15,6 +15,7 @@
 from arctic.tickstore.toplevel import TopLevelTickStore, TickStoreLibrary
 from dateutil.rrule import rrule, DAILY
 from arctic.tickstore.tickstore import TickStore
+from arctic.exceptions import UnhandledDtypeException
 
 
 def test_raise_exception_if_daterange_is_not_provided():
@@ -168,3 +169,10 @@ def test_read():
                                                     sentinel.include_columns, include_images=sentinel.include_images),
                                                call(sentinel.symbol, tsl.date_range.intersection.return_value,
                                                     sentinel.include_columns, include_images=sentinel.include_images)]
+
+
+def test_slice_raises():
+    m = TopLevelTickStore(Mock())
+    with pytest.raises(UnhandledDtypeException) as e:
+        m._slice("abc", 1, 2)
+    assert("Can't persist type" in str(e))