Add test

tests/e2e/test_e2e.py

@@ -712,3 +712,23 @@ async def simulate_network_latency(request: httpx.Request) -> httpx.Response:
     run_test_suite(
         id="my-test-id", test_cases=test_cases, evaluators=[MyEvaluator()], fn=test_fn, max_test_case_concurrency=100
     )
+
+
+def test_get_dataset():
+    dataset = client.get_dataset("test-dataset", "1")
+    assert dataset.revision_id == "cm1mgsnx1000bf9f85p99kx3g"
+    assert dataset.name == "test-dataset"
+    assert dataset.schema_version == "1"
+    assert len(dataset.items) == 1
+    assert dataset.items[0].splits == ["test-split"]
+    assert dataset.items[0].data == {"Test Property": "Test Value 2"}
+
+
+def test_get_dataset_by_revision_id():
+    dataset = client.get_dataset("test-dataset", "1", revision_id="cm1mgsgu30006f9f85zhuwzlx")
+    assert dataset.revision_id == "cm1mgsgu30006f9f85zhuwzlx"
+    assert dataset.name == "test-dataset"
+    assert dataset.schema_version == "1"
+    assert len(dataset.items) == 1
+    assert dataset.items[0].splits == ["test-split"]
+    assert dataset.items[0].data == {"Test Property": "Test Value"}