def test_format_instances_response(self):
features = ['a_var', 'c_var', 'd_var']
header, datarows = format_instances(self.instances, features)
self.assertEqual(header, ['a_var', 'c_var', 'd_var'])
self.assertEqual(datarows, [[
'aval',
'cval',
'dval',
] for _ in range(len(self.instances))])
def test_format_instances_response_nonexistantfeature(self):
with pytest.raises(AttributeError):
format_instances(self.instances, ['robot_not_a_real_feature'])
def test_format_instances_response_nofeatures(self):
header, datarows = format_instances(self.instances, [])
assert header == []
assert datarows == [[] for _ in range(len(self.instances))]
def test_format_instances_response_noinstances(self):
features = ['a_var']
header, datarows = format_instances([], features)
assert header == features
assert datarows == []
def test_format_instances_response(self):
features = ['a_var', 'c_var', 'd_var']
header, datarows = format_instances(self.instances, features)
assert header == ['a_var', 'c_var', 'd_var']
assert datarows == [['aval', 'cval', 'dval']
for _ in range(len(self.instances))]
def test_format_instances_response_nofeatures(self):
header, datarows = format_instances(self.instances, [])
self.assertEqual(header, [])
self.assertEqual(datarows, [[] for _ in range(len(self.instances))])
def test_format_instances_response_noinstances(self):
features = ['a_var']
header, datarows = format_instances([], features)
self.assertEqual(header, features)
self.assertEqual(datarows, [])
