Browse Source

[beeswax] Clean up test_partition_values

Erick Tryzelaar 11 years ago
parent
commit
c195f0e69c
1 changed files with 5 additions and 2 deletions
  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({})