Total Complexity | 129 |
Total Lines | 405 |
Duplicated Lines | 0 % |
Complex classes like Orange.tests.sql.FilterStringTest often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
1 | import unittest |
||
317 | @sql_test |
||
318 | class FilterStringTest(PostgresTest): |
||
319 | def setUp(self): |
||
320 | self.data = [ |
||
321 | [w] for w in "Lorem ipsum dolor sit amet, consectetur adipiscing" |
||
322 | "elit. Vestibulum vel dolor nulla. Etiam elit lectus, mollis nec" |
||
323 | "mattis sed, pellentesque in turpis. Vivamus non nisi dolor. Etiam" |
||
324 | "lacinia dictum purus, in ullamcorper ante vulputate sed. Nullam" |
||
325 | "congue blandit elementum. Donec blandit laoreet posuere. Proin" |
||
326 | "quis augue eget tortor posuere mollis. Fusce vestibulum bibendum" |
||
327 | "neque at convallis. Donec iaculis risus volutpat malesuada" |
||
328 | "vehicula. Ut cursus tempor massa vulputate lacinia. Pellentesque" |
||
329 | "eu tortor sed diam placerat porttitor et volutpat risus. In" |
||
330 | "vulputate rutrum lacus ac sagittis. Suspendisse interdum luctus" |
||
331 | "sem auctor commodo.".split(' ')] + [[None], [None]] |
||
332 | self.conn, self.table_name = self.create_sql_table(self.data) |
||
333 | self.table = SqlTable(self.conn, self.table_name) |
||
334 | |||
335 | def tearDown(self): |
||
336 | self.drop_sql_table(self.table_name) |
||
337 | |||
338 | def test_filter_string_is_defined(self): |
||
339 | filtered_data = filter.Values(conditions=[ |
||
340 | filter.FilterString(-1, filter.FilterString.IsDefined) |
||
341 | ])(self.table) |
||
342 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
343 | for row in self.data if row[0] is not None] |
||
344 | |||
345 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
346 | self.assertSequenceEqual(filtered_data, correct_data) |
||
347 | |||
348 | def test_filter_string_equal(self): |
||
349 | filtered_data = filter.Values(conditions=[ |
||
350 | filter.FilterString(-1, filter.FilterString.Equal, 'in') |
||
351 | ])(self.table) |
||
352 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
353 | for row in self.data if row[0] == 'in'] |
||
354 | |||
355 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
356 | self.assertSequenceEqual(filtered_data, correct_data) |
||
357 | |||
358 | def test_filter_string_equal_case_insensitive_value(self): |
||
359 | filtered_data = filter.Values(conditions=[ |
||
360 | filter.FilterString(-1, filter.FilterString.Equal, 'In', |
||
361 | case_sensitive=False) |
||
362 | ])(self.table) |
||
363 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
364 | for row in self.data if row[0] == 'in'] |
||
365 | |||
366 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
367 | self.assertSequenceEqual(filtered_data, correct_data) |
||
368 | |||
369 | def test_filter_string_equal_case_insensitive_data(self): |
||
370 | filtered_data = filter.Values(conditions=[ |
||
371 | filter.FilterString(-1, filter.FilterString.Equal, 'donec', |
||
372 | case_sensitive=False) |
||
373 | ])(self.table) |
||
374 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
375 | for row in self.data if row[0] == 'Donec'] |
||
376 | |||
377 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
378 | self.assertSequenceEqual(filtered_data, correct_data) |
||
379 | |||
380 | def test_filter_string_not_equal(self): |
||
381 | filtered_data = filter.Values(conditions=[ |
||
382 | filter.FilterString(-1, filter.FilterString.NotEqual, 'in') |
||
383 | ])(self.table) |
||
384 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
385 | for row in self.data if row[0] != 'in'] |
||
386 | |||
387 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
388 | self.assertSequenceEqual(filtered_data, correct_data) |
||
389 | |||
390 | def test_filter_string_not_equal_case_insensitive_value(self): |
||
391 | filtered_data = filter.Values(conditions=[ |
||
392 | filter.FilterString(-1, filter.FilterString.NotEqual, 'In', |
||
393 | case_sensitive=False) |
||
394 | ])(self.table) |
||
395 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
396 | for row in self.data if row[0] != 'in'] |
||
397 | |||
398 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
399 | self.assertSequenceEqual(filtered_data, correct_data) |
||
400 | |||
401 | def test_filter_string_not_equal_case_insensitive_data(self): |
||
402 | filtered_data = filter.Values(conditions=[ |
||
403 | filter.FilterString(-1, filter.FilterString.NotEqual, 'donec', |
||
404 | case_sensitive=False) |
||
405 | ])(self.table) |
||
406 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
407 | for row in self.data if row[0] != 'Donec'] |
||
408 | |||
409 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
410 | self.assertSequenceEqual(filtered_data, correct_data) |
||
411 | |||
412 | def test_filter_string_less(self): |
||
413 | filtered_data = filter.Values(conditions=[ |
||
414 | filter.FilterString(-1, filter.FilterString.Less, 'A') |
||
415 | ])(self.table) |
||
416 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
417 | for row in self.data |
||
418 | if row[0] is not None and row[0] < 'A'] |
||
419 | |||
420 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
421 | self.assertSequenceEqual(filtered_data, correct_data) |
||
422 | |||
423 | def test_filter_string_less_case_insensitive_value(self): |
||
424 | filtered_data = filter.Values(conditions=[ |
||
425 | filter.FilterString(-1, filter.FilterString.Less, 'In', |
||
426 | case_sensitive=False) |
||
427 | ])(self.table) |
||
428 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
429 | for row in self.data |
||
430 | if row[0] is not None and row[0].lower() < 'in'] |
||
431 | |||
432 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
433 | self.assertSequenceEqual(filtered_data, correct_data) |
||
434 | |||
435 | def test_filter_string_less_case_insensitive_data(self): |
||
436 | filtered_data = filter.Values(conditions=[ |
||
437 | filter.FilterString(-1, filter.FilterString.Less, 'donec', |
||
438 | case_sensitive=False) |
||
439 | ])(self.table) |
||
440 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
441 | for row in self.data |
||
442 | if row[0] is not None and row[0].lower() < 'donec'] |
||
443 | |||
444 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
445 | self.assertSequenceEqual(filtered_data, correct_data) |
||
446 | |||
447 | def test_filter_string_less_equal(self): |
||
448 | filtered_data = filter.Values(conditions=[ |
||
449 | filter.FilterString(-1, filter.FilterString.LessEqual, 'A') |
||
450 | ])(self.table) |
||
451 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
452 | for row in self.data |
||
453 | if row[0] is not None and row[0] <= 'A'] |
||
454 | |||
455 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
456 | self.assertSequenceEqual(filtered_data, correct_data) |
||
457 | |||
458 | def test_filter_string_less_equal_case_insensitive_value(self): |
||
459 | filtered_data = filter.Values(conditions=[ |
||
460 | filter.FilterString(-1, filter.FilterString.LessEqual, 'In', |
||
461 | case_sensitive=False) |
||
462 | ])(self.table) |
||
463 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
464 | for row in self.data |
||
465 | if row[0] is not None and row[0].lower() <= 'in'] |
||
466 | |||
467 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
468 | self.assertSequenceEqual(filtered_data, correct_data) |
||
469 | |||
470 | def test_filter_string_less_equal_case_insensitive_data(self): |
||
471 | filtered_data = filter.Values(conditions=[ |
||
472 | filter.FilterString(-1, filter.FilterString.LessEqual, 'donec', |
||
473 | case_sensitive=False) |
||
474 | ])(self.table) |
||
475 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
476 | for row in self.data |
||
477 | if row[0] is not None and row[0].lower() <= 'donec'] |
||
478 | |||
479 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
480 | self.assertSequenceEqual(filtered_data, correct_data) |
||
481 | |||
482 | def test_filter_string_greater(self): |
||
483 | filtered_data = filter.Values(conditions=[ |
||
484 | filter.FilterString(-1, filter.FilterString.Greater, 'volutpat') |
||
485 | ])(self.table) |
||
486 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
487 | for row in self.data |
||
488 | if row[0] is not None and row[0] > 'volutpat'] |
||
489 | |||
490 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
491 | self.assertSequenceEqual(filtered_data, correct_data) |
||
492 | |||
493 | def test_filter_string_greater_case_insensitive_value(self): |
||
494 | filtered_data = filter.Values(conditions=[ |
||
495 | filter.FilterString(-1, filter.FilterString.Greater, 'In', |
||
496 | case_sensitive=False) |
||
497 | ])(self.table) |
||
498 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
499 | for row in self.data |
||
500 | if row[0] is not None and row[0].lower() > 'in'] |
||
501 | |||
502 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
503 | self.assertSequenceEqual(filtered_data, correct_data) |
||
504 | |||
505 | def test_filter_string_greater_case_insensitive_data(self): |
||
506 | filtered_data = filter.Values(conditions=[ |
||
507 | filter.FilterString(-1, filter.FilterString.Greater, 'donec', |
||
508 | case_sensitive=False) |
||
509 | ])(self.table) |
||
510 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
511 | for row in self.data |
||
512 | if row[0] is not None and row[0].lower() > 'donec'] |
||
513 | |||
514 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
515 | self.assertSequenceEqual(filtered_data, correct_data) |
||
516 | |||
517 | def test_filter_string_greater_equal(self): |
||
518 | filtered_data = filter.Values(conditions=[ |
||
519 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'volutpat') |
||
520 | ])(self.table) |
||
521 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
522 | for row in self.data |
||
523 | if row[0] is not None and row[0] >= 'volutpat'] |
||
524 | |||
525 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
526 | self.assertSequenceEqual(filtered_data, correct_data) |
||
527 | |||
528 | def test_filter_string_greater_equal_case_insensitive_value(self): |
||
529 | filtered_data = filter.Values(conditions=[ |
||
530 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'In', |
||
531 | case_sensitive=False) |
||
532 | ])(self.table) |
||
533 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
534 | for row in self.data |
||
535 | if row[0] is not None and row[0].lower() >= 'in'] |
||
536 | |||
537 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
538 | self.assertSequenceEqual(filtered_data, correct_data) |
||
539 | |||
540 | def test_filter_string_greater_equal_case_insensitive_data(self): |
||
541 | filtered_data = filter.Values(conditions=[ |
||
542 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'donec', |
||
543 | case_sensitive=False) |
||
544 | ])(self.table) |
||
545 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
546 | for row in self.data |
||
547 | if row[0] is not None and row[0].lower() >= 'donec'] |
||
548 | |||
549 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
550 | self.assertSequenceEqual(filtered_data, correct_data) |
||
551 | |||
552 | def test_filter_string_between(self): |
||
553 | filtered_data = filter.Values(conditions=[ |
||
554 | filter.FilterString(-1, filter.FilterString.Between, 'a', 'c') |
||
555 | ])(self.table) |
||
556 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
557 | for row in self.data |
||
558 | if row[0] is not None and 'a' <= row[0] <= 'c'] |
||
559 | |||
560 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
561 | self.assertSequenceEqual(filtered_data, correct_data) |
||
562 | |||
563 | def test_filter_string_between_case_insensitive_value(self): |
||
564 | filtered_data = filter.Values(conditions=[ |
||
565 | filter.FilterString(-1, filter.FilterString.Between, 'I', 'O', |
||
566 | case_sensitive=False) |
||
567 | ])(self.table) |
||
568 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
569 | for row in self.data |
||
570 | if row[0] is not None and 'i' < row[0].lower() <= 'o'] |
||
571 | |||
572 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
573 | self.assertSequenceEqual(filtered_data, correct_data) |
||
574 | |||
575 | def test_filter_string_between_case_insensitive_data(self): |
||
576 | filtered_data = filter.Values(conditions=[ |
||
577 | filter.FilterString(-1, filter.FilterString.Between, 'i', 'O', |
||
578 | case_sensitive=False) |
||
579 | ])(self.table) |
||
580 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
581 | for row in self.data |
||
582 | if row[0] is not None and 'i' <= row[0].lower() <= 'o'] |
||
583 | |||
584 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
585 | self.assertSequenceEqual(filtered_data, correct_data) |
||
586 | |||
587 | def test_filter_string_contains(self): |
||
588 | filtered_data = filter.Values(conditions=[ |
||
589 | filter.FilterString(-1, filter.FilterString.Contains, 'et') |
||
590 | ])(self.table) |
||
591 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
592 | for row in self.data |
||
593 | if row[0] is not None and 'et' in row[0]] |
||
594 | |||
595 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
596 | self.assertSequenceEqual(filtered_data, correct_data) |
||
597 | |||
598 | def test_filter_string_contains_case_insensitive_value(self): |
||
599 | filtered_data = filter.Values(conditions=[ |
||
600 | filter.FilterString(-1, filter.FilterString.Contains, 'eT', |
||
601 | case_sensitive=False) |
||
602 | ])(self.table) |
||
603 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
604 | for row in self.data |
||
605 | if row[0] is not None and 'et' in row[0].lower()] |
||
606 | |||
607 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
608 | self.assertSequenceEqual(filtered_data, correct_data) |
||
609 | |||
610 | def test_filter_string_contains_case_insensitive_data(self): |
||
611 | filtered_data = filter.Values(conditions=[ |
||
612 | filter.FilterString(-1, filter.FilterString.Contains, 'do', |
||
613 | case_sensitive=False) |
||
614 | ])(self.table) |
||
615 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
616 | for row in self.data |
||
617 | if row[0] is not None and 'do' in row[0].lower()] |
||
618 | |||
619 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
620 | self.assertSequenceEqual(filtered_data, correct_data) |
||
621 | |||
622 | def test_filter_string_outside(self): |
||
623 | filtered_data = filter.Values(conditions=[ |
||
624 | filter.FilterString(-1, filter.FilterString.Outside, 'am', 'di') |
||
625 | ])(self.table) |
||
626 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
627 | for row in self.data |
||
628 | if row[0] is not None and not 'am' < row[0] < 'di'] |
||
629 | |||
630 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
631 | self.assertSequenceEqual(filtered_data, correct_data) |
||
632 | |||
633 | def test_filter_string_outside_case_insensitive(self): |
||
634 | filtered_data = filter.Values(conditions=[ |
||
635 | filter.FilterString(-1, filter.FilterString.Outside, 'd', 'k', |
||
636 | case_sensitive=False) |
||
637 | ])(self.table) |
||
638 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
639 | for row in self.data |
||
640 | if row[0] is not None and not 'd' < row[0].lower() < 'k'] |
||
641 | |||
642 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
643 | self.assertSequenceEqual(filtered_data, correct_data) |
||
644 | |||
645 | def test_filter_string_starts_with(self): |
||
646 | filtered_data = filter.Values(conditions=[ |
||
647 | filter.FilterString(-1, filter.FilterString.StartsWith, 'D') |
||
648 | ])(self.table) |
||
649 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
650 | for row in self.data |
||
651 | if row[0] is not None and row[0].startswith('D')] |
||
652 | |||
653 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
654 | self.assertSequenceEqual(filtered_data, correct_data) |
||
655 | |||
656 | def test_filter_string_starts_with_case_insensitive(self): |
||
657 | filtered_data = filter.Values(conditions=[ |
||
658 | filter.FilterString(-1, filter.FilterString.StartsWith, 'D', |
||
659 | case_sensitive=False) |
||
660 | ])(self.table) |
||
661 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
662 | for row in self.data |
||
663 | if row[0] is not None |
||
664 | and row[0].lower().startswith('d')] |
||
665 | |||
666 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
667 | self.assertSequenceEqual(filtered_data, correct_data) |
||
668 | |||
669 | def test_filter_string_ends_with(self): |
||
670 | filtered_data = filter.Values(conditions=[ |
||
671 | filter.FilterString(-1, filter.FilterString.EndsWith, 's') |
||
672 | ])(self.table) |
||
673 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
674 | for row in self.data |
||
675 | if row[0] is not None and row[0].endswith('s')] |
||
676 | |||
677 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
678 | self.assertSequenceEqual(filtered_data, correct_data) |
||
679 | |||
680 | def test_filter_string_ends_with_case_insensitive(self): |
||
681 | filtered_data = filter.Values(conditions=[ |
||
682 | filter.FilterString(-1, filter.FilterString.EndsWith, 'S', |
||
683 | case_sensitive=False) |
||
684 | ])(self.table) |
||
685 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
686 | for row in self.data |
||
687 | if row[0] is not None |
||
688 | and row[0].lower().endswith('s')] |
||
689 | |||
690 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
691 | self.assertSequenceEqual(filtered_data, correct_data) |
||
692 | |||
693 | def test_filter_string_list(self): |
||
694 | filtered_data = filter.Values(conditions=[ |
||
695 | filter.FilterStringList(-1, ['et', 'in']) |
||
696 | ])(self.table) |
||
697 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
698 | for row in self.data if row[0] in ['et', 'in']] |
||
699 | |||
700 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
701 | self.assertSequenceEqual(filtered_data, correct_data) |
||
702 | |||
703 | def test_filter_string_list_case_insensitive_value(self): |
||
704 | filtered_data = filter.Values(conditions=[ |
||
705 | filter.FilterStringList(-1, ['Et', 'In'], case_sensitive=False) |
||
706 | ])(self.table) |
||
707 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
708 | for row in self.data if row[0] in ['et', 'in']] |
||
709 | |||
710 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
711 | self.assertSequenceEqual(filtered_data, correct_data) |
||
712 | |||
713 | def test_filter_string_list_case_insensitive_data(self): |
||
714 | filtered_data = filter.Values(conditions=[ |
||
715 | filter.FilterStringList(-1, ['donec'], case_sensitive=False) |
||
716 | ])(self.table) |
||
717 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
718 | for row in self.data if row[0] in ['Donec']] |
||
719 | |||
720 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
721 | self.assertSequenceEqual(filtered_data, correct_data) |
||
722 |