From c88412e3f1ac28eb6affb8ffc4caa5b9f8d710e7 Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Fri, 25 Sep 2015 16:42:25 -0400 Subject: [PATCH] TST: Add coverage for long and short count. Make basic checks of position stats test all fields, by adding coverage for `shorts_count` and `longs_count`. --- tests/test_perf_tracking.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_perf_tracking.py b/tests/test_perf_tracking.py index 72319220..dcb0bab9 100644 --- a/tests/test_perf_tracking.py +++ b/tests/test_perf_tracking.py @@ -2239,10 +2239,12 @@ class TestPositionTracker(unittest.TestCase): pos_stats = position_tracker.calc_position_stats(pt) self.assertEqual(100, pos_stats.long_value) self.assertEqual(100 + 300000, pos_stats.long_exposure) + self.assertEqual(2, pos_stats.longs_count) # Test short-only methods self.assertEqual(-200, pos_stats.short_value) self.assertEqual(-200 - 400000, pos_stats.short_exposure) + self.assertEqual(2, pos_stats.shorts_count) # Test gross and net values self.assertEqual(100 + 200, pos_stats.gross_value)