From c86d221b290ecf6da8bbe6f58555ca755c155194 Mon Sep 17 00:00:00 2001 From: Greg Quinn <greg.quinn@ssec.wisc.edu> Date: Wed, 4 Oct 2017 11:06:10 -0500 Subject: [PATCH] Refactor rdrgen tests --- tests/test_rdrgen.py | 55 +++++++++++++++++++++++++------------------- 1 file changed, 31 insertions(+), 24 deletions(-) diff --git a/tests/test_rdrgen.py b/tests/test_rdrgen.py index 2cd701e..beb6fef 100644 --- a/tests/test_rdrgen.py +++ b/tests/test_rdrgen.py @@ -1,7 +1,7 @@ -from datetime import datetime import os import subprocess from StringIO import StringIO +from datetime import datetime import h5py @@ -9,30 +9,9 @@ import edosl0util.rdrgen as m from edosl0util.jpssrdr import decode_rdr_blob from edosl0util.stream import jpss_packet_stream -snpp_base_time = 1698019234000000 # from CDFCB vol I "Time of First Ascending Node" table -cris_gran_len = 31997000 - - -def test_calc_iet_granule(): - run = lambda t: m.calc_iet_granule(snpp_base_time, cris_gran_len, t) - gran = 1880240293174000 - prev_gran = 1880240261177000 - next_gran = 1880240325171000 - assert run(gran) == gran - assert run(gran + 1) == gran - assert run(gran - 1) == prev_gran - assert run(gran + cris_gran_len) == next_gran - -def test_calc_iet_aggregate(): - grans_per_aggr = 15 - run = lambda t: m.calc_iet_aggregate(snpp_base_time, cris_gran_len, grans_per_aggr, t) - aggr = 1880240037198000 - aggr_len = grans_per_aggr * cris_gran_len - assert run(aggr - 1) == aggr - aggr_len - assert run(aggr) == aggr - assert run(aggr + aggr_len - 1) == aggr - assert run(aggr + aggr_len) == aggr + aggr_len +class TestViirsGroupedPacketTimeTracker(object): + pass def test_can_reproduce_rdr_from_class(): @@ -68,3 +47,31 @@ def test_can_reproduce_rdr_from_class(): stdout=subprocess.PIPE, stderr=subprocess.STDOUT) assert p.communicate()[0] == '' assert p.returncode == 0 + + +class TestGranulation(object): + def test_calc_iet_granule(self): + def run(t): + return m.calc_iet_granule(self.snpp_base_time, self.cris_gran_len, t) + gran = 1880240293174000 + prev_gran = 1880240261177000 + next_gran = 1880240325171000 + assert run(gran) == gran + assert run(gran + 1) == gran + assert run(gran - 1) == prev_gran + assert run(gran + self.cris_gran_len) == next_gran + + def test_calc_iet_aggregate(self): + grans_per_aggr = 15 + def run(t): + return m.calc_iet_aggregate( + self.snpp_base_time, self.cris_gran_len, grans_per_aggr, t) + aggr = 1880240037198000 + aggr_len = grans_per_aggr * self.cris_gran_len + assert run(aggr - 1) == aggr - aggr_len + assert run(aggr) == aggr + assert run(aggr + aggr_len - 1) == aggr + assert run(aggr + aggr_len) == aggr + aggr_len + + snpp_base_time = 1698019234000000 # CDFCB vol I "Time of First Ascending Node" table + cris_gran_len = 31997000 -- GitLab