def test_record_data(self):
r = Record(zip(["name", "age", "married"], ["Alice", 33, True]))
self.assertEqual(r.data(), {"name": "Alice", "age": 33, "married": True})
self.assertEqual(r.data("name"), {"name": "Alice"})
self.assertEqual(r.data("age", "name"), {"age": 33, "name": "Alice"})
self.assertEqual(r.data("age", "name", "shoe size"), {"age": 33, "name": "Alice", "shoe size": None})
self.assertEqual(r.data(0, "name"), {"name": "Alice"})
self.assertEqual(r.data(0), {"name": "Alice"})
self.assertEqual(r.data(1, 0), {"age": 33, "name": "Alice"})
with self.assertRaises(IndexError):
_ = r.data(1, 0, 999)
def test_record_data():
r = Record(zip(["name", "age", "married"], ["Alice", 33, True]))
assert r.data() == {"name": "Alice", "age": 33, "married": True}
assert r.data("name") == {"name": "Alice"}
assert r.data("age", "name") == {"age": 33, "name": "Alice"}
assert r.data("age", "name", "shoe size") == {"age": 33, "name": "Alice", "shoe size": None}
assert r.data(0, "name") == {"name": "Alice"}
assert r.data(0) == {"name": "Alice"}
assert r.data(1, 0) == {"age": 33, "name": "Alice"}
with pytest.raises(IndexError):
_ = r.data(1, 0, 999)
