From 60fc558f7ba9e5b6d76feb4aa644682f95d86bf5 Mon Sep 17 00:00:00 2001
From: Timofey Solomko <tsolomko@gmail.com>
Date: Sun, 5 Feb 2017 19:16:30 +0300
Subject: [PATCH] Test updated with last changes to api.

---
 Tests/DeflateTests.swift | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/Tests/DeflateTests.swift b/Tests/DeflateTests.swift
index 71b7f15b..cffed929 100644
--- a/Tests/DeflateTests.swift
+++ b/Tests/DeflateTests.swift
@@ -76,10 +76,11 @@ class DeflateTests: XCTestCase {
             return
         }
 
-        let answerBytes = answerData.toArray(type: UInt8.self)
-        let bldCodes = Deflate.lengthEncode(answerBytes)
-        let huffmanEncodedBytes = Deflate.huffmanEncode(bldCodes)
-        let reUncompData = try! Deflate.decompress(compressedData: Data(bytes: huffmanEncodedBytes))
+        guard let deflatedData = try? Deflate.compress(data: answerData) else {
+            XCTFail("Unable to deflate data")
+            return
+        }
+        let reUncompData = try! Deflate.decompress(compressedData: deflatedData)
         print(String(data: reUncompData, encoding: .utf8)!)
     }