def generic_bug(self, files, join=False): with TemporaryDirectory() as f: args, data = bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) if join: files = [ 'data/test/source/personal/[email protected]_%s.fit' % file for file in files ] args, data = bootstrap_dir(f, mm(DEV), 'read', *files) read(args, data) else: for file in files: args, data = bootstrap_dir( f, mm(DEV), 'read', 'data/test/source/personal/[email protected]_%s.fit' % file) read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: # steps summary = s.query(StatisticJournal).join(StatisticName). \ filter(StatisticJournal.time >= local_date_to_time('2018-10-07'), StatisticJournal.time < local_date_to_time('2018-10-08'), StatisticName.owner == MonitorCalculator, StatisticName.name == N.DAILY_STEPS).one() # connect has 3031 for this date. self.assertEqual(summary.value, 3031)
def test_constant(self): with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), 'read', 'data/test/source/personal/2018-03-04-qdp.fit') read(args, data) with data.db.session_context() as s: stats = Statistics(s, activity_journal='2018-03-04 07:16:33', with_timespan=True). \ by_name(SegmentReader, N.LATITUDE, N.LONGITUDE, N.SPHERICAL_MERCATOR_X, N.SPHERICAL_MERCATOR_Y, N.DISTANCE, N.ELEVATION, N.SPEED, N.CADENCE, N.HEART_RATE).df stats.describe() sepn = pd.Series(stats.index).diff().median() # 7 secs start = stats.index.min() # 2018-03-04 10:16:33+00:00 finish = stats.index.max() # 2018-03-04 16:34:51+00:00 even = pd.DataFrame({'keep': True}, index=pd.date_range(start=start, end=finish, freq=sepn)) both = stats.join(even, how='outer', sort=True) both.interpolate(method='index', limit_area='inside', inplace=True) both = both.loc[both['keep'] == True].drop(columns=['keep']) both = both.loc[both[N.TIMESPAN_ID].isin( stats[N.TIMESPAN_ID].unique())] both.describe()
def generic_bug(self, files): with NamedTemporaryFile() as f: args, sys, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) for file in files: args, sys, db = bootstrap_file( f, m(V), '5', mm(DEV), 'monitor', mm(FAST), 'data/test/source/personal/[email protected]_%s.fit' % file) monitor(args, sys, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(sys, db, PipelineType.STATISTIC, force=True, start='2018-01-01', n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: # steps summary = s.query(StatisticJournal).join(StatisticName). \ filter(StatisticJournal.time >= local_date_to_time('2018-10-07'), StatisticJournal.time < local_date_to_time('2018-10-08'), StatisticName.owner == MonitorCalculator, StatisticName.name == DAILY_STEPS).one() # connect has 3031 for this date. self.assertEqual(summary.value, 3031)
def test_values(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) for file in ('24696157869', '24696160481', '24696163486'): args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'monitor', mm(FAST), 'data/test/source/personal/[email protected]_%s.fit' % file) monitor(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, force=True, start='2018-01-01', n_cpu=1) run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: mjournals = s.query(MonitorJournal).order_by(MonitorJournal.start).all() assert mjournals[2].start == to_time('2018-09-06 15:06:00'), mjournals[2].start print(mjournals[2].fit_file) # steps summary = s.query(StatisticJournal).join(StatisticName). \ filter(StatisticJournal.time >= local_date_to_time('2018-09-06'), StatisticJournal.time < local_date_to_time('2018-09-07'), StatisticName.owner == MonitorCalculator, StatisticName.name == DAILY_STEPS).one() # connect has 12757 for this date, self.assertEqual(summary.value, 12757) # heart rate summary = s.query(StatisticJournal).join(StatisticName). \ filter(StatisticJournal.time >= local_date_to_time('2018-09-06'), StatisticJournal.time < local_date_to_time('2018-09-07'), StatisticName.owner == MonitorCalculator, StatisticName.name == REST_HR).one() self.assertEqual(summary.value, 45)
def spline(self, smooth=0): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', 'constants', '--set', SRTM1_DIR, '/home/andrew/archive/srtm1') constants(args, db) with db.session_context() as s: yield spline_elevation_from_constant(log, s, smooth=smooth)
def spline(self, smooth=0): with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir(f, m(V), '5', 'constants', 'set', SRTM1_DIR_CNAME, '/home/andrew/archive/srtm1', mm(FORCE)) constants(args, data) with data.db.session_context() as s: yield spline_elevation_from_constant(s, smooth=smooth)
def bilinear(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, sys, db = bootstrap_file(f, m(V), '5', 'constants', 'set', SRTM1_DIR, '/home/andrew/archive/srtm1') constants(args, sys, db) with db.session_context() as s: yield bilinear_elevation_from_constant(s)
def test_context(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) with db.session_context() as s: with Timestamp(owner=TestTimestamp).on_success(s): n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 0) n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 1)
def test_florian(self): with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir(f, m(V), '5', mm(DEV), 'read', 'data/test/source/private/florian.fit') read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: self.__assert_basic_stats(s)
def test_920(self): for src in '920xt-2019-05-16_19-42-54.fit', '920xt-2019-05-16_19-42-54.fit': with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir(f, m(V), '5', mm(DEV), 'read', f'data/test/source/other/{src}') read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: self.__assert_basic_stats(s)
def test_set(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) with db.session_context() as s: source = add(s, Source()) n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 0) Timestamp.set(s, TestTimestamp, source=source) n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 1) t = s.query(Timestamp).filter(Timestamp.owner == TestTimestamp).one() self.assertAlmostEqual(t.time.timestamp(), dt.datetime.now().timestamp(), 1)
def test_michael(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'activities', mm(FAST), 'data/test/source/other/2019-05-09-051352-Running-iWatchSeries3.fit') activities(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: self.__assert_basic_stats(s)
def test_920(self): for src in '920xt-2019-05-16_19-42-54.fit', '920xt-2019-05-16_19-42-54.fit': with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'activities', mm(FAST), f'data/test/source/other/{src}') activities(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: self.__assert_basic_stats(s)
def test_constant(self): with TemporaryDirectory() as f: args, data = bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) with data.db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 14) args, data = bootstrap_dir(f, m(V), '5', 'constants', 'set', 'fthr:%', '154', mm(FORCE)) constants(args, data) with data.db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 14)
def test_constant(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) with db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 10) args, db = bootstrap_file(f, m(V), '5', 'constants', '--set', 'FTHR.%', '154') constants(args, db) with db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 10)
def test_florian(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'activities', mm(FAST), 'data/test/source/private/florian.fit') activities(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: self.__assert_basic_stats(s)
def test_michael(self): with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), 'read', 'data/test/source/other/2019-05-09-051352-Running-iWatchSeries3.fit' ) read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: self.__assert_basic_stats(s)
def test_constant(self): # this is currently failing because there are multiple statistics called Active Distance with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) with db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 13) args, db = bootstrap_file(f, m(V), '5', 'constants', '--set', 'FTHR.%', '154') constants(args, db) with db.session_context() as s: n = s.query(count(Constant.id)).scalar() self.assertEqual(n, 13)
def test_context_error(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) with db.session_context() as s: try: with Timestamp(owner=TestTimestamp).on_success(s): n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 0) raise Exception('foo') except Exception as e: self.assertEqual(str(e), 'foo') # for some weird reason assertRaisesRegex was not working n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 0)
def test_context(self): with TemporaryDirectory() as f: args, data = bootstrap_dir(f, m(V), '5') args, data = bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) with data.db.session_context() as s: with Timestamp(owner=TestTimestamp).on_success(s): n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 0) n = s.query(count(Timestamp.id)).scalar() self.assertEqual(n, 1)
def test_models(self): with TemporaryDirectory() as f: args, data = bootstrap_dir(f, m(V), '5', configurator=default) with data.db.session_context() as s: start(s, 'bike', 'cotic', '2018-01-01', True) start(s, 'bike', 'marin', '2018-01-01', False) change(s, 'cotic', 'chain', 'sram', None, True, True) change(s, 'cotic', 'chain', 'kcm', '2018-01-01', False, False) change(s, 'cotic', 'chain', 'sram', '2018-05-01', False, False) change(s, 'cotic', 'chain', 'kcm', '2018-07-01', False, False) change(s, 'cotic', 'chain', 'sram', '2018-04-01', False, False) start(s, 'bike', 'bowman', '2018-01-01', False) change(s, 'bowman', 'chain', 'sram', None, False, True) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), READ, 'data/test/source/personal/2018-08-03-rec.fit', m(K), 'cotic') read(args, data) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), READ, 'data/test/source/personal/2018-08-27-rec.fit', m(K), 'cotic') read(args, data) run_pipeline(data, PipelineType.CALCULATE, like=['%Activity%'], n_cpu=1) run_pipeline(data, PipelineType.CALCULATE, like=['%Kit%'], n_cpu=1) with data.db.session_context() as s: bike = get_name(s, 'bike').to_model(s, depth=3, statistics=INDIVIDUAL, own_models=False) self.assertEqual(bike[TYPE], KitGroup.SIMPLE_NAME) self.assertEqual(bike[NAME], 'bike') self.assertEqual(len(bike[ITEMS]), 3) cotic = [ item for item in bike[ITEMS] if item[NAME] == 'cotic' ][0] self.assertEqual(cotic[TYPE], KitItem.SIMPLE_NAME) self.assertEqual(cotic[NAME], 'cotic') self.assertEqual(len(cotic[COMPONENTS]), 1) chain = cotic[COMPONENTS][0] self.assertEqual(chain[TYPE], KitComponent.SIMPLE_NAME) self.assertEqual(chain[NAME], 'chain') self.assertEqual(len(chain[MODELS]), 6) self.assertFalse(STATISTICS in bike)
def test_empty_data(self): with NamedTemporaryFile() as f: args, sys, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, sys, db = bootstrap_file( f, m(V), '5', mm(DEV), 'monitor', mm(FAST), 'data/test/source/other/37140810636.fit') monitor(args, sys, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(sys, db, PipelineType.STATISTIC, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: n = s.query(func.count(StatisticJournal.id)).scalar() self.assertEqual(n, 20) mjournal = s.query(MonitorJournal).one() self.assertNotEqual(mjournal.start, mjournal.finish)
def test_heart_alarms(self): with TemporaryDirectory() as f: bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), 'read', 'data/test/source/personal/2016-07-19-mpu-s-z2.fit') read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: for stat in s.query(StatisticJournal). \ join(StatisticName). \ filter(StatisticName.name == N.ACTIVE_DISTANCE).all(): self.assertGreater(stat.value, 30)
def test_monitor(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'monitor', mm(FAST), 'data/test/source/personal/25822184777.fit') monitor(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, force=True, start='2018-01-01') # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: n = s.query(func.count(StatisticJournal.id)).scalar() # self.assertEqual(n, 111) self.assertEqual(n, 108) # why? mjournal = s.query(MonitorJournal).one() self.assertNotEqual(mjournal.start, mjournal.finish)
def test_empty_data(self): with TemporaryDirectory() as f: args, data = bootstrap_dir(f, m(V), '5') bootstrap_dir(f, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir( f, m(V), '5', mm(DEV), 'read', 'data/test/source/other/37140810636.fit') read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: n = s.query(func.count(StatisticJournal.id)).scalar() self.assertEqual(n, 44) mjournal = s.query(MonitorJournal).one() self.assertNotEqual(mjournal.start, mjournal.finish)
def test_heart_alarms(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5') bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, db = bootstrap_file(f, m(V), '5', mm(DEV), 'activities', mm(FAST), 'data/test/source/personal/2016-07-19-mpu-s-z2.fit') activities(args, db) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(db, PipelineType.STATISTIC, n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with db.session_context() as s: stat = s.query(StatisticJournal). \ join(StatisticName). \ filter(StatisticName.name == ACTIVE_DISTANCE).one() self.assertGreater(stat.value, 30000)
def test_constant(self): with NamedTemporaryFile() as f: bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) args, sys, db = bootstrap_file(f, m(V), '5', 'constants', 'set', 'FTHR.%', '154') constants(args, sys, db) args, sys, db = bootstrap_file(f, m(V), '5', 'constants', 'show', 'FTHR.%') constants(args, sys, db) args, sys, db = bootstrap_file( f, m(V), '5', mm(DEV), 'activities', mm(FAST), 'data/test/source/personal/2018-03-04-qdp.fit') activities(args, sys, db) with db.session_context() as s: stats = activity_statistics(s, LATITUDE, LONGITUDE, SPHERICAL_MERCATOR_X, SPHERICAL_MERCATOR_Y, DISTANCE, ELEVATION, SPEED, CADENCE, HEART_RATE, local_time='2018-03-04 07:16:33', activity_group_name='Bike', with_timespan=True) stats.describe() sepn = pd.Series(stats.index).diff().median() # 7 secs start = stats.index.min() # 2018-03-04 10:16:33+00:00 finish = stats.index.max() # 2018-03-04 16:34:51+00:00 even = pd.DataFrame({'keep': True}, index=pd.date_range(start=start, end=finish, freq=sepn)) both = stats.join(even, how='outer', sort=True) both.interpolate(method='index', limit_area='inside', inplace=True) both = both.loc[both['keep'] == True].drop(columns=['keep']) both = both.loc[both[TIMESPAN_ID].isin( stats[TIMESPAN_ID].unique())] both.describe()
def test_segment_bug(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5', mm(DEV), configurator=default) paths = ['/home/andrew/archive/fit/bike/2016-07-27-pm-z4.fit'] run_pipeline(db, PipelineType.ACTIVITY, paths=paths, force=True)
def test_activities(self): with TemporaryDirectory() as base: bootstrap_dir(base, m(V), '5') bootstrap_dir(base, m(V), '5', mm(DEV), configurator=default) args, data = bootstrap_dir(base, m(V), '5', 'constants', 'set', 'SRTM1.dir', '/home/andrew/archive/srtm1', mm(FORCE)) constants(args, data) args, data = bootstrap_dir( base, m(V), '5', mm(DEV), 'read', 'data/test/source/personal/2018-08-27-rec.fit') read(args, data) # run('sqlite3 %s ".dump"' % f.name, shell=True) run_pipeline(data, PipelineType.CALCULATE, force=True, start='2018-01-01', n_cpu=1) # run('sqlite3 %s ".dump"' % f.name, shell=True) with data.db.session_context() as s: n_raw = s.query(count(StatisticJournalFloat.id)). \ join(StatisticName). \ filter(StatisticName.name == N.RAW_ELEVATION).scalar() self.assertEqual(2099, n_raw) n_fix = s.query(count(StatisticJournalFloat.id)). \ join(StatisticName). \ filter(StatisticName.name == N.ELEVATION).scalar() self.assertEqual(2099, n_fix) # WHY does this jump around? n = s.query(count(StatisticJournal.id)).scalar() # self.assertEqual(50403, n) self.assertTrue(n > 30000) self.assertTrue(n < 100000) journal = s.query(ActivityJournal).one() self.assertNotEqual(journal.start, journal.finish)
def test_bikes(self): with NamedTemporaryFile() as f: args, db = bootstrap_file(f, m(V), '5', configurator=default) with db.session_context() as s: with self.assertRaises(Exception) as ctx: start(s, 'bike', 'cotic', None, False) self.assertTrue('--force' in str(ctx.exception), ctx.exception) start(s, 'bike', 'cotic', None, True) # run('sqlite3 %s ".dump"' % f.name, shell=True) with self.assertRaises(Exception) as ctx: start(s, 'xxxx', 'marin', None, False) self.assertTrue('--force' in str(ctx.exception), ctx.exception) start(s, 'bike', 'marin', None, False) with self.assertRaises(Exception) as ctx: change(s, 'cotic', 'chain', 'sram', '2018-02-01', False, False) self.assertTrue('--force' in str(ctx.exception)) change(s, 'cotic', 'chain', 'sram', None, True, True) change(s, 'cotic', 'chain', 'kcm', '2018-01-01', False, False) change(s, 'cotic', 'chain', 'sram', '2018-05-01', False, False) change(s, 'cotic', 'chain', 'kcm', '2018-07-01', False, False) change(s, 'cotic', 'chain', 'sram', '2018-04-01', False, False) with self.assertRaises(Exception) as ctx: start(s, 'bike', 'bike', None, True) self.assertTrue('bike' in str(ctx.exception), ctx.exception) with self.assertRaises(Exception) as ctx: start(s, 'bike', 'sram', None, True) self.assertTrue('sram' in str(ctx.exception), ctx.exception) start(s, 'bike', 'bowman', None, False) change(s, 'bowman', 'chain', 'sram', None, False, True) self.assertEqual(len(show(s, 'cotic', None)), 3) self.assertEqual(len(statistics(s, 'bike')), 16) self.assertEqual(len(statistics(s, 'cotic')), 20) self.assertEqual(len(statistics(s, 'chain')), 33) self.assertEqual(len(statistics(s, 'sram')), 16) finish(s, 'bowman', None, False) with self.assertRaises(Exception) as ctx: finish(s, 'bowman', None, False) self.assertTrue('retired' in str(ctx.exception), ctx.exception) self.assertEqual( len( KitModel.get_all(s, KitItem.get(s, 'cotic'), KitComponent.get(s, 'chain'))), 5) undo(s, 'cotic', 'chain', 'sram', None, True) self.assertEqual( len( KitModel.get_all(s, KitItem.get(s, 'cotic'), KitComponent.get(s, 'chain'))), 2) undo(s, 'cotic', 'chain', 'kcm', None, True) self.assertEqual( len( KitModel.get_all(s, KitItem.get(s, 'cotic'), KitComponent.get(s, 'chain'))), 0) undo(s, 'bowman', 'chain', 'sram', None, True) self.assertFalse(KitComponent.get(s, 'chain', require=False))