def test_grid_getitem():
g = Grid()
g.column['test'] = {}
row = {'test': 'This is a test'}
g.append(row)
assert g[0] is row
def test_grid_setitem():
g = Grid()
g.column['test'] = {}
row_1 = {'test': 'This is a test'}
row_2 = {'test': 'This is another test'}
g.append(row_1)
g[0] = row_2
assert g[0] is row_2
def test_grid_append_nover_list():
g = Grid()
assert g.version == Version('2.0')
g.column['test'] = {}
row_1 = {'test': 'This is a test'}
row_2 = {'test': ['This should fail']}
g.append(row_1)
assert g.version == Version('2.0')
g.append(row_2)
assert g.version == Version('3.0')
def test_grid_append_v2_list_fail():
g = Grid(version='2.0')
g.column['test'] = {}
row_1 = {'test': 'This is a test'}
row_2 = {'test': ['This should fail']}
g.append(row_1)
try:
g.append(row_2)
assert False, 'Appended invalid data type'
except ValueError as e:
assert str(e) == 'Data type requires version 3.0'
def test_slice():
grid = Grid(columns={'id': {}, 'site': {}})
grid.append({
'id': 'id1',
})
grid.append({'id': 'id2'})
grid.append({'id': 'id3'})
result = grid[0:2]
assert len(result) == 2
assert result['id1']
assert result['id2']
def test_grid_append_notdict():
g = Grid()
g.column['test'] = {}
row_1 = {'test': 'This is a test'}
row_2 = 'This is not a dict'
g.append(row_1)
try:
g.append(row_2)
assert False
except TypeError as e:
assert str(e) == 'value must be a dict'
assert len(g) == 1
def test_grid_setitem_notdict():
g = Grid()
g.column['test'] = {}
row = {'test': 'This is a test'}
g.append(row)
try:
g[0] = 'This is not a dict'
assert False, 'Accepted a string'
except TypeError:
pass
assert len(g) == 1
assert g[0]['test'] == 'This is a test'
def test_filter_and_limit():
grid = Grid(columns={'id': {}, 'site': {}})
grid.append({
'id': 'id1',
'site': MARKER,
'equip': 'Chicago',
'geoPostalCode': "78280",
'ahu': MARKER,
'geoCity': 'Chicago',
'curVal': 76
})
grid.append({
'id': 'id2',
'hvac': MARKER,
'geoPostalCode': "23220",
'curVal': 75
})
grid.append({
'equip': 'Chicago',
'hvac': MARKER,
'siteRef': Ref('id1'),
'curVal': 74
})
assert len(grid.filter('not acme', limit=1)) == 1
def test_grid_filter():
grid = Grid(
columns={
'id': {},
'site': {},
'equip': {},
'geoPostalCode': {},
'ahu': {},
'geoCity': {},
'curVal': {},
'hvac': {},
'siteRef': {}
})
grid.append({
'id': 'id1',
'site': MARKER,
'equip': 'Chicago',
'geoPostalCode': "78280",
'ahu': MARKER,
'geoCity': 'Chicago',
'curVal': 76
})
grid.append({
'id': 'id2',
'hvac': MARKER,
'geoPostalCode': "23220",
'curVal': 75
})
grid.append({
'equip': 'Chicago',
'hvac': MARKER,
'siteRef': Ref('id1'),
'curVal': 74
})
result = grid.filter('site')
assert len(result) == 1
assert result['id1']
result = grid.filter('equip == "Chicago"')
assert len(result) == 2
assert result[0]['id'] == 'id1'
assert result[1]['equip'] == 'Chicago'
result = grid.filter('not id')
assert len(result) == 1
assert result[0]['equip'] == 'Chicago'
def test_slide_get():
grid = Grid(
columns={
'id': {},
'site': {},
'equip': {},
'geoPostalCode': {},
'ahu': {},
'geoCity': {},
'curVal': {},
'hvac': {},
'siteRef': {}
})
grid.append({
'id': 'id1',
'site': MARKER,
'equip': 'Chicago',
'geoPostalCode': "78280",
'ahu': MARKER,
'geoCity': 'Chicago',
'curVal': 76
})
grid.append({
'id': 'id2',
'hvac': MARKER,
'geoPostalCode': "23220",
'curVal': 75
})
grid.append({
'equip': 'Chicago',
'hvac': MARKER,
'siteRef': Ref('id1'),
'curVal': 74
})
assert len(grid[0:1]) == 1
assert len(grid[1:]) == 2
assert len(grid[:]) == 3
def test_grid_specification_filter_sample():
grid = Grid(
columns={
'id': {},
'site': {},
'equip': {},
'geoPostalCode': {},
'ahu': {},
'geoCity': {},
'curVal': {},
'hvac': {},
'siteRef': {}
})
grid.append({
'id': 'id1',
'site': MARKER,
'equip': 'Chicago',
'geoPostalCode': "78280",
'ahu': MARKER,
'geoCity': 'Chicago',
'curVal': 76
})
grid.append({
'id': 'id2',
'hvac': MARKER,
'geoPostalCode': "23220",
'curVal': 75
})
grid.append({
'equip': 'Chicago',
'hvac': MARKER,
'siteRef': Ref('id1'),
'curVal': 74
})
# specification samples
result = grid.filter('geoPostalCode == "23220"')
assert len(result) == 1
assert result[0]['id'] == 'id2'
result = grid.filter('geoPostalCode != "23220"')
assert len(result) == 1
assert result[0]['id'] == 'id1'
result = grid.filter('curVal < 75')
assert len(result) == 1
assert result[0]['siteRef'] == Ref('id1')
result = grid.filter('curVal <= 75')
assert len(result) == 2
assert result[0]['id'] == 'id2'
assert result[1]['siteRef'] == Ref('id1')
result = grid.filter('curVal > 75')
assert len(result) == 1
assert result[0]['id'] == 'id1'
result = grid.filter('curVal >= 75')
assert len(result) == 2
assert result[0]['id'] == 'id1'
assert result[1]['id'] == 'id2'
result = grid.filter('site or equip')
assert len(result) == 2
assert result[0]['id'] == 'id1'
assert result[1]['siteRef'] == Ref('id1')
result = grid.filter('equip and hvac')
assert len(result) == 1
assert result[0]['siteRef'] == Ref('id1')
result = grid.filter('equip and not ahu')
assert len(result) == 1
assert result[0]['siteRef'] == Ref('id1')
result = grid.filter('equip and siteRef->geoCity == "Chicago"')
assert len(result) == 1
assert result[0]['equip'] == 'Chicago'
