Skip to content

Commit cb17e32

Browse files
committed
correct unit test
1 parent 1646864 commit cb17e32

File tree

5 files changed

+10
-6
lines changed

5 files changed

+10
-6
lines changed

src/ToonEncoder.Generator/ToonEncoderGenerator.cs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,10 @@ static void EmitTabularArrayConverter(SourceProductionContext sourceProductionCo
7474
}));
7575

7676
var source = $$"""
77+
// <auto-generated/>
78+
#pragma warning disable
79+
#nullable enable
80+
7781
using Cysharp.AI.Internal;
7882
using System;
7983
using System.Buffers;

src/ToonEncoder/ToonEncoder.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,13 @@ public static void Encode<TBufferWriter>(ref TBufferWriter bufferWriter, JsonEle
3939
{
4040
var toonWriter = ToonWriter.Create(ref bufferWriter);
4141
Encode(ref toonWriter, element);
42+
toonWriter.Flush();
4243
}
4344

4445
public static void Encode<TBufferWriter>(ref ToonWriter<TBufferWriter> toonWriter, JsonElement element)
4546
where TBufferWriter : IBufferWriter<byte>
4647
{
4748
WriteElement(ref toonWriter, element);
48-
toonWriter.Flush();
4949
}
5050

5151
public static byte[] EncodeToUtf8Bytes(JsonElement element)

tests/ToonEncoder.Tests/FixtureTest.cs

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,26 +1,23 @@
11
using Cysharp.AI;
22
using System.Buffers;
3-
using System.Runtime.InteropServices;
43
using System.Text;
54
using System.Text.Json;
6-
using System.Xml.Linq;
7-
using TUnit.Core;
85

96
namespace ToonEncoder.Tests;
107

118
// https://github.com/toon-format/spec/tree/main/tests
129

1310
public class FixtureTest
1411
{
15-
public record EncodeTestData(TestCase Fixture)
12+
public record struct EncodeTestData(TestCase Fixture)
1613
{
1714
public override string ToString()
1815
{
1916
return Fixture.Name;
2017
}
2118
}
2219

23-
public async Task<EncodeTestData[]> LoadTestCases(string filePath)
20+
public static async Task<EncodeTestData[]> LoadTestCases(string filePath)
2421
{
2522
var fullPath = Path.Combine(Directory.GetCurrentDirectory(), filePath);
2623
var json = await File.ReadAllTextAsync(fullPath);
@@ -111,6 +108,7 @@ public async Task Delimiters(EncodeTestData testData)
111108
var toonWriter = ToonWriter.Create(ref bufferWriter, delimiter);
112109

113110
Cysharp.AI.ToonEncoder.Encode(ref toonWriter, testData.Fixture.Input);
111+
toonWriter.Flush();
114112

115113
var toon = Encoding.UTF8.GetString(bufferWriter.WrittenSpan);
116114
await Assert.That(toon).IsEqualTo(testData.Fixture.Expected);

tests/ToonEncoder.Tests/ToonEncoderEncodeAsTabularArrayTests.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ public async Task EncodeAsTabularArray_ToonWriter_ProducesTabularLayout()
5555
var toonWriter = ToonWriter.Create(ref bufferWriter);
5656

5757
Cysharp.AI.ToonEncoder.EncodeAsTabularArray(ref toonWriter, element);
58+
toonWriter.Flush();
5859

5960
var result = Encoding.UTF8.GetString(bufferWriter.WrittenSpan);
6061
await Assert.That(result).IsEqualTo(ExpectedTabular);

tests/ToonEncoder.Tests/ToonEncoderEncodeTests.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ public async Task Encode_WithToonWriter_FlushesResult()
5454
var toonWriter = ToonWriter.Create(ref bufferWriter);
5555

5656
Cysharp.AI.ToonEncoder.Encode(ref toonWriter, element);
57+
toonWriter.Flush();
5758

5859
var result = Encoding.UTF8.GetString(bufferWriter.WrittenSpan);
5960
await Assert.That(result).IsEqualTo(ExpectedToon);

0 commit comments

Comments
 (0)