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)