From 4e16456eb39a937279257c56909dabed447fce33 Mon Sep 17 00:00:00 2001
From: aouinizied <aouinizied@gmail.com>
Date: Fri, 18 Oct 2019 23:37:25 +0200
Subject: [PATCH] Improve tests.

---
 nfstream/observer.py | 2 +-
 tests.py             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/nfstream/observer.py b/nfstream/observer.py
index eda5d6e..245019d 100644
--- a/nfstream/observer.py
+++ b/nfstream/observer.py
@@ -76,4 +76,4 @@ class Observer:
         if self.packet_generator is not None:
             for timestamp, packet in self.packet_generator:
                 packet_information = process_packet(timestamp, packet)
-                yield packet_information
\ No newline at end of file
+                yield packet_information
diff --git a/tests.py b/tests.py
index 4bbb706..213f9f9 100644
--- a/tests.py
+++ b/tests.py
@@ -47,7 +47,7 @@ class TestMethods(unittest.TestCase):
                             capacity=128000,
                             inactive_timeout=maxsize,
                             active_timeout=maxsize)
-        exports = streamer.exports
+        exports = list(streamer)
         exports_ground_truth = []
         del streamer
         self.assertEqual(exports, exports_ground_truth)