소스 검색

[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({})