Fix more tests

Author: AdrienVannson

betterproto2/tests/inputs/proto3_field_presence_oneof/test_proto3_field_presence_oneof.py

@@ -1,8 +1,4 @@
-from tests.output_betterproto.proto3_field_presence_oneof import (
-    Nested,
-    Test,
-    WithOptional,
-)
+from tests.outputs.proto3_field_presence_oneof.proto3_field_presence_oneof import Nested, Test, WithOptional
 
 
 def test_serialization():

betterproto2/tests/inputs/regression_387/test_regression_387.py

@@ -1,4 +1,4 @@
-from tests.output_betterproto.regression_387 import ParentElement, Test
+from tests.outputs.regression_387.regression_387 import ParentElement, Test
 
 
 def test_regression_387():

betterproto2/tests/inputs/regression_414/test_regression_414.py

@@ -1,4 +1,4 @@
-from tests.output_betterproto.regression_414 import Test
+from tests.outputs.regression_414.regression_414 import Test
 
 
 def test_full_cycle():

betterproto2/tests/inputs/repeated_duration_timestamp/test_repeated_duration_timestamp.py

@@ -1,9 +1,6 @@
-from datetime import (
-    datetime,
-    timedelta,
-)
+from datetime import datetime, timedelta
 
-from tests.output_betterproto.repeated_duration_timestamp import Test
+from tests.outputs.repeated_duration_timestamp.repeated_duration_timestamp import Test
 
 
 def test_roundtrip():

betterproto2/tests/inputs/timestamp_dict_encode/test_timestamp_dict_encode.py

@@ -6,7 +6,7 @@
 
 import pytest
 
-from tests.output_betterproto.timestamp_dict_encode import Test
+from tests.outputs.timestamp_dict_encode.timestamp_dict_encode import Test
 
 # Current World Timezone range (UTC-12 to UTC+14)
 MIN_UTC_OFFSET_MIN = -12 * 60

betterproto2_compiler/tests/generate.py

@@ -84,7 +84,7 @@ async def generate_test(
 
 async def main_async():
     # Don't compile too many tests in parallel
-    semaphore = asyncio.Semaphore(4)
+    semaphore = asyncio.Semaphore(os.cpu_count() or 1)
 
     tasks = [
         generate_test("any", semaphore),
@@ -115,13 +115,18 @@ async def main_async():
         generate_test("oneof_default_value_serialization", semaphore),
         generate_test("pickling", semaphore),
         generate_test("proto3_field_presence", semaphore),
+        generate_test("proto3_field_presence_oneof", semaphore),
         generate_test("recursivemessage", semaphore),
+        generate_test("regression_387", semaphore),
+        generate_test("regression_414", semaphore),
         generate_test("repeated", semaphore),
         generate_test("repeatedpacked", semaphore),
+        generate_test("repeated_duration_timestamp", semaphore),
         generate_test("service", semaphore),
         generate_test("service_uppercase", semaphore),
         generate_test("simple_service", semaphore),
         generate_test("stream_stream", semaphore),
+        generate_test("timestamp_dict_encode", semaphore),
     ]
     await asyncio.gather(*tasks)