浏览代码

[beeswax] Clean up test_partition_values

Erick Tryzelaar 11 年之前
父节点
当前提交
c195f0e
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 5 2
      apps/beeswax/src/beeswax/tests.py

+ 5 - 2
apps/beeswax/src/beeswax/tests.py

@@ -1792,8 +1792,11 @@ class TestHiveServer2API():
   def test_partition_values(self):
     table = MockHiveServerTable({'path_location': '/my/table'})
 
-    assert_equal(['2013022516'], PartitionValueCompatible(['datehour=2013022516'], table).values)
-    assert_equal(['2011-07', '2011-07-01', '12'], PartitionValueCompatible(['month=2011-07/dt=2011-07-01/hr=12'], table).values)
+    value = PartitionValueCompatible(['datehour=2013022516'], table)
+    assert_equal(['2013022516'], value.values)
+
+    value = PartitionValueCompatible(['month=2011-07/dt=2011-07-01/hr=12'], table)
+    assert_equal(['2011-07', '2011-07-01', '12'], value.values)
 
   def test_table_properties(self):
     table = MockHiveServerTable({})