From 55cc3456f3ac16ca6fe296f495440dc78d70ea26 Mon Sep 17 00:00:00 2001
From: Yoshi Askharoun <jjask7@gmail.com>
Date: Thu, 8 Feb 2024 13:50:51 -0600
Subject: [PATCH] Add type import and export to parser test

---
 libs/UIX.Test/Assembly.cs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/libs/UIX.Test/Assembly.cs b/libs/UIX.Test/Assembly.cs
index 905a283..7a3004f 100644
--- a/libs/UIX.Test/Assembly.cs
+++ b/libs/UIX.Test/Assembly.cs
@@ -15,8 +15,12 @@ public class Assembly(ITestOutputHelper output)
     {
         const string code =
 """
+.export main 0 UI
+
 .import-ns Me as me
 .import-ns assembly://UIX/Microsoft.Iris as iris
+.import-type iris:Command
+
 
 .section data
 
@@ -36,8 +40,8 @@ main:
         output.WriteLine(ast.ToString());
 
         Assert.NotNull(ast);
-        Assert.Equal(2, ast.Imports.Count());
-        Assert.Equal(9 + 2, ast.Body.Count());
+        Assert.Equal(3 + 1 + 2, ast.Directives.Count());
+        Assert.Equal(9, ast.Body.Count());
     }
 
     [Fact]