def test_nested_data(self): data = [{ "customer": { "name": "John", "address": { "street": "Street 1", "number": "42" } }, "product": { "sku": "1", "price": 9.99 } }, { "customer": { "name": "Bob", "address": { "street": "Street 2", "number": "314" } }, "product": { "sku": "2", "price": 15.00 } }] self.assertEqual(extract_header(data), [ 'customer.address.number', 'customer.address.street', 'customer.name', 'product.price', 'product.sku' ])
def test_simple_data(self): data = [{ 'key_1': 'value 1', 'key_2': 'value 2' }, { 'key_1': 'value 3', 'key_2': 'value 4' }] self.assertEqual(extract_header(data), ['key_1', 'key_2'])
def test_simple_data(self): data = [ { "key_1": "value 1", "key_2": "value 2" }, { "key_1": "value 3", "key_2": "value 4" }, ] self.assertEqual(extract_header(data), ["key_1", "key_2"])
def test_interrupt_if_keys_dont_change(self): data = [{"key": "value"} for _ in range(100)] self.assertEqual(extract_header(data), ["key"])
def test_interrupt_if_keys_dont_change(self): data = [{'key': 'value'} for _ in range(100)] self.assertEqual(extract_header(data), ['key'])