update test

ohmayr · ohmayr · commit 4879d532a46e · 2026-01-30T23:22:41.000Z
diff --git a/tests/unit/samplegen/test_samplegen.py b/tests/unit/samplegen/test_samplegen.py
@@ -397,7 +397,7 @@ def test_get_sample_imports_with_request_from_another_package(
     imports = samplegen._get_sample_imports(sample, rpc)
 
     assert imports == [
-        "from a.b import c_pb2  # type: ignore",
+        "import a.b.c_pb2 as c_pb2  # type: ignore",
         "from mollusc.cephalopod import teuthida_v1",
     ]
 

Original file line number	Diff line number	Diff line change
`@@ -397,7 +397,7 @@ def test_get_sample_imports_with_request_from_another_package(`
`397`	`397`	`imports = samplegen._get_sample_imports(sample, rpc)`
`398`	`398`
`399`	`399`	`assert imports == [`
`400`		`- "from a.b import c_pb2 # type: ignore",`
	`400`	`+ "import a.b.c_pb2 as c_pb2 # type: ignore",`
`401`	`401`	`"from mollusc.cephalopod import teuthida_v1",`
`402`	`402`	`]`
`403`	`403`